我为什么用if语句获取“'numpy.ndarray'对象不可调用”

Question

我是编程新手，我想在我希望“ y”函数与x的特定条件不同的情况下绘制图。 这就是我试图做到的方式：

import matplotlib.pyplot as plt

import numpy as np

x= np.arange(0,10,0.1)

if x(x<5):
    y=x
else:
    y=0

plt.plot(x,y)
plt.show()

Answer 1

plt.plot(x, [x_ if x_ < 5 else 0 for x_ in x])

plot方法将x与y绘制为线条和/或标记。 使用列表推导基于x数组的值创建y数组

y = [x_ if x_ < 5 else 0 for x_ in x]

完全一样

y = list()
for i in range(len(x)):
   if x[i] < 5:
     y.append(x[i])
   else:
     y.append(0)

Answer 2

该错误源自x(x<5) 。 Python将第一个x解析为一个函数，并尝试将其作为参数x<5调用，从而触发错误，因为x不是您可以调用的函数，它是numpy.ndarray ，因此解释了numpy.ndarray object is not callable错误。

尝试这个：

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(0,10,0.1)
y = [(xx if xx<5 else 0) for xx in x]

plt.plot(x,y)
plt.show()