我想用Matplotlib在Python中绘制以下从0到5的分段函数。
f(x) = 1, x != 2; f(x) = 0, x = 2
在Python中。。。
def f(x):
if(x == 2): return 0
else: return 1
使用NumPy创建一个数组
x = np.arange(0., 5., 0.2)
array([ 0. , 0.2, 0.4, 0.6, 0.8, 1. , 1.2, 1.4, 1.6, 1.8, 2. ,
2.2, 2.4, 2.6, 2.8, 3. , 3.2, 3.4, 3.6, 3.8, 4. , 4.2,
4.4, 4.6, 4.8])
我试过像。。。
import matplotlib.pyplot as plt
plt.plot(x,f(x))
或者。。。
vecfunc = np.vectorize(f)
result = vecfunc(t)
或者。。。
def piecewise(x):
if x == 2: return 0
else: return 1
import matplotlib.pyplot as plt
x = np.arange(0., 5., 0.2)
plt.plot(x, map(piecewise, x))
ValueError: x and y must have same first dimension
但我没有正确使用这些函数,现在只是随机猜测如何做到这一点。
一些答案开始出现。。。但是这些点被连接成一条曲线。我们怎么画出点呢?
问题是函数
f
不接受数组作为输入,而是接受单个数字。你可以:函数
map
接受一个函数f
,一个数组x
,并返回另一个数组,其中函数f
应用于数组的每个元素。可以对数组使用np.piecewise:
相关问题 更多 >
编程相关推荐