python matplotlib.pyplot 和 numpy 问题

Question

我有以下问题。我想评估以下功能

def sigLinZ(self,val,omega):
    A = 1
    C = 0 
    D = 1 

    B =1./(omega*val*1j)

    return np.asmatrix(np.array([[A,B],[C,D]]))

以这样一种方式，我可以在 pyplot 中使用它:

omega = numpy.arange(0,100,1)
y = classInstance.sigLinZ(12,omega)
plt.plot(omega,y)

但这行不通。 Python 说:

Traceback (most recent call last):


File "testImpedanz.py", line 132, in test6_lineImpedanz
    print "neue Matrix: ",  lineImpe.sigLinZ('C',lineImpe.C(),np.array([600e6,300e6]))
  File "/afs/physnet.uni-hamburg.de/users/ap_h/pgwozdz/Dokumente/PythonSkriptsPHD/ImpedanzCalculation.py", line 350, in sigLinZ
    return np.mat(np.array([[A,B],[C,D]]))
TypeError: only length-1 arrays can be converted to Python scalars

我知道对于 numpy 函数这个过程工作正常但对于我的函数它根本不起作用。

Answer 1

您正试图将一个数组插入到传递给方法的 omega 定义中的矩阵元素中。您需要遍历 omega 将每个元素分别传递给 sigLinZ，或者您需要重写 sigLinZ 来处理数组并返回一些内容像矩阵列表或 3D 数组。