Python - ошибка при рисовании контурного графика с использованием scipy.integrate.quad

Я пытаюсь нарисовать контурный график на Python, используя немного более сложный код, чем приведенный ниже. Но та же проблема возникает в следующем простом примере:

import numpy as np
from scipy import integrate
import matplotlib.pyplot as plt

def A(x):
    return integrate.quad(lambda i: x-i, 0, x)

n = 100
x = np.linspace(0, 10, n)
y = np.linspace(0, 10, n)

X, Y = np.meshgrid(x, y)
M = A(X) + Y  # THE ERROR ARISES HERE

plt.contour(X, Y, M)
plt.show()

Следующее сообщение об ошибке отображается в строке, указанной выше:

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

Хочу использовать integrate.quad().

Может кто-нибудь мне помочь? Заранее спасибо.

Две проблемы: (1) по умолчанию quad возвращает значения два (см. строка документации); (2) пределы интегрирования должны быть скалярами, а не массивами.

— 02.06.2018 06:53

@WarrenWeckesser Спасибо за ваш комментарий.

— 02.06.2018 13:14

@ Mr.T Я искал фигуру, подобную той, что была в ответе на ImportanceOfBeingEarnest. Спасибо, в любом случае.

— 02.06.2018 13:16

python matplotlib scipy contour

02.06.2018 06:16

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

167

Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Как правильно указал @WarrenWeckesser в комментариях, две проблемы:

quad возвращает кортеж, из которого вам понадобится только первый член.
Интеграл, который вы хотите вычислить, является скалярным, поэтому вы не можете передать ему 2D-массив.

Решения:

проиндексируйте возврат quad: quad( ..., )[0], чтобы извлечь только первый элемент
vectorize вызов A, так что он вызывается со всеми скалярами массива X один за другим. Обратите внимание, что это не очень эффективно, но должно подойти для массива из 100x100 элементов, представленного здесь.

Полное решение могло бы выглядеть как

import numpy as np
from scipy import integrate
import matplotlib.pyplot as plt

def A(x):
    return integrate.quad(lambda i: x-i, 0, x)[0]

n = 100
x = np.linspace(0, 10, n)
y = np.linspace(0, 10, n)

X, Y = np.meshgrid(x, y)
M = np.vectorize(A)(X) + Y

plt.contour(X, Y, M)
plt.show()