python - 绘制 error_only size-1 数组可以转换为 Python 标量

Question

我想绘制一个函数，每次都会收到此错误:只有 size-1 数组可以转换为 Python 标量。我想在 w 上绘制我的函数。

代码是:

import matplotlib.pyplot as plt
import numpy as np
import math
a = 10;
w = np.arange (0.001,10, 0.1)

function = (1/(np.pi*(w**2))+ np.pi)*abs(((-2*math.atan(a))*w)-(2*math.atanh(((w**2) + 1)/(-1*(w**2) + 2*(a**2) +1))*w)+ (np.pi*w) -(2*math.atanh(w/(a))))

plt.plot(w, function) 

plt.show() 

感谢您的宝贵时间。

Answer 1

您最好使用 numpy.arctan 和 numpy.arctanh 而不是 math 等效项，因为它们可以处理数组:

a = 10;
w = np.arange (0.001,10, 0.1)

function = (1/(np.pi*(w**2))+ np.pi)*abs(((-2*np.arctan(a))*w)-(2*np.arctanh(((w**2) + 1)/(-1*(w**2) + 2*(a**2) +1))*w)+ (np.pi*w) -(2*np.arctanh(w/(a))))

plt.plot(w, function) 

plt.show()