Я пытаюсь построить простую функцию в python (x + sqrt(x^2 + 2x)). Вот мой код:
import pylab as pl
import numpy as np
import math
X = np.linspace(-999999,999999)
Y = (X+math.sqrt(X**2+2*X))
pl.plot(X,Y)
pl.show()
Вот ошибка, с которой я сталкиваюсь: TypeError: только массивы длины -1 могут быть преобразованы в скаляры Python
использовать
Y = [(x+math.sqrt(x**2+2*x)) for x in X]
or
Y = map(lambda x: (x+math.sqrt(x**2+2*x)), X)
для создания списка значений y.
Вы также можете векторизовать свою функцию и применить ее. Дополнительные замечания см. в разделе Понимание списка, карта и производительность numpy.vectorize. .
Проблема с вашим подходом заключается в том, что функция math.sqrt ожидает одно число в качестве входных данных. Поскольку у вас есть массив numpy с более чем одним элементом, вы должны использовать встроенную функцию numpy sqrt, как это было предложено Рори Йорком, т.е.
Y = X+np.sqrt(X**2+2*X
Поскольку функция numpy гораздо более оптимизирована, она обеспечивает лучшую производительность, например:
X = np.arange(10000)
timeit Y = map(lambda x: (x+math.sqrt(x**2+2*x)), X)
10 loops, best of 3: 82.1 ms per loop
timeit Y = X+np.sqrt(X**2+2*X)
10000 loops, best of 3: 108 µs per loop
Y = X+np.sqrt(X**2+2*X)? 07.06.2014