python numpy数组画图_python – Numpy和Pyplot的条件绘图

我试图绘制一个有条件定义的函数.特别：

U(x)=(2 ** delta)/((Dd)** delta)*(D / 2 – (x-x0))** delta,abs(x-x0)小于D / 2和0除此以外.

但我的问题是我希望将x,x0作为numpy数组,因为这是我在其余实际代码中使用它们的方式.

我已经设置了以下示例：

import numpy as np

import matplotlib.pyplot as plt

AD = 0.2

D = 0.4

delta = 8

def Parabolic(x,delta,D,AD):

x0 = np.round(x)

tempx = np.abs(x-x0)

tempD = D/2*np.ones(len(x))

if tempx

return ((2**delta)/(D-AD)**delta)*(D/2 - (x-x0))**delta

else:

return 0

figure = plt.figure(figsize=(10,8),dpi=72)

xmin = -1.0

xmax = 1.0

X = np.linspace(xmin,xmax,1000)

plt.plot(X,Parabolic(X,delta=8,D=0.4,AD=0.2))

显然,这个例子不起作用,因为行tempx< tempD引起了列表的真值 - 值不明确的错误. 我搜索了numpy的文档并找到了函数np.less(tempx,tempD).但如果我取代tempx