小编典典

TypeError:只有长1的数组可以转换为Python标量,而图显示

python

我有这样的Python代码:

import numpy as np
import matplotlib.pyplot as plt

def f(x):
    return np.int(x)

x = np.arange(1, 15.1, 0.1)
plt.plot(x, f(x))
plt.show()

而这样的错误:

TypeError: only length-1 arrays can be converted to Python scalars

我该如何解决?


阅读 214

收藏
2020-12-20

共1个答案

小编典典

当函数需要单个值但您传递数组时,会出现错误“仅将length-1数组可以转换为Python标量”。

如果查看的呼叫签名np.int,您会看到它接受单个值,而不是数组。通常,如果要对数组中的每个元素应用一个接受单个元素的函数,则可以使用np.vectorize

import numpy as np
import matplotlib.pyplot as plt

def f(x):
    return np.int(x)
f2 = np.vectorize(f)
x = np.arange(1, 15.1, 0.1)
plt.plot(x, f2(x))
plt.show()

您可以跳过f(x)的定义,而只需将np.int传递给vectorize函数:f2 = np.vectorize(np.int)

注意,这np.vectorize只是一个便利函数,基本上是for循环。在大型阵列上这将是低效的。每当你有可能,使用真正的矢量函数或方法(如astype(int)为@FFT建议。

2020-12-20