只有长度为 1 的数组可以转换为 Python 标量

Question

如何更改该代码以进行运行？必须运行在实轴和虚轴上产生函数w(omega) = 1/(1 + 1j*omega) 的图形。

import matplotlib.pyplot as plt
import numpy as np


def func(a):
    for x in range(len(a)):
        plt.plot([0, a[x].real], [0, a[x].imag], 'ro-', label='python')
    limit=np.max(np.ceil(np.absolute(a))) # set limits for axis
    plt.xlim((-limit,limit))
    plt.ylim((-limit,limit))
    plt.ylabel('Imaginary')
    plt.xlabel('Real')
    plt.show()

omega = np.linspace(-4, 4, 251)
a = np.arange(1) + 1j*np.arange(omega, 1)  
func(a)

Answer 1

回答您的具体问题：错误源于omega 是一个数组，但arange 需要标量参数：从到步骤的值。由于它得到一个长度为 1 的数组 (omega)，它无法确定应该选择 omega 的哪个值作为范围的起点。因此出现错误：omega 无法转换为标量（这将使arange 工作）。

但目前尚不清楚您为什么要以这种方式绘制，以及您想在数组a 中放入什么。