绘制 error_only 大小为 1 的数组可以转换为 Python 标量

Question

我想绘制一个函数，每次出现此错误时：只有大小为 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() 

谢谢你的时间。