python - matplotlib中的步进函数

我已经看到有关matplotlib中的步函数的一些问题，但这是不同的。
这是我的功能：

def JerkFunction(listOfJerk):
    '''Return the plot of a sequence of jerk'''
    #initialization of the jerk
    x = np.linspace(0,5,4)
    y = listOfJerk #step signal

    plt.axis([0,5,-2,2])
    plt.step(x,y,'y') #step display
    plt.xlabel('Time (s)')
    plt.ylabel('Jerk (m/s^3)')

    plt.title('Jerk produced by the engine')

    return plt.show()

我希望获得在放入JerkFunction([1,1,-1,1])时获得的曲线，但是通过输入：[1,-1,1,-1]，实际上，在一开始，在实际情况下，加加速度值是0，而在t=0处，它变成了jerk=+1，然后在t= 1处是Jerk=-1等。

最佳答案

我认为您在这个问题Matlibplot step function index 0上遇到了同样的问题。您遇到的问题与步骤在何处更改相对于x值（doc）的值有关。

下面演示了它可以执行此操作的三种方法。为了清楚起见，曲线垂直移动。水平虚线为“零”，垂直虚线为x值。

x = np.linspace(0,5,3)
y = np.array([1,-1,1])

fig = plt.figure()
ax = fig.add_subplot(111)
ax.step(x,y,color='r',label='pre')
ax.step(x,y+3,color='b',label='post',where='post')
ax.step(x,y+6,color='g',label='mid',where='mid')
for j in [0,3,6]:
    ax.axhline(j,color='k',linestyle='--')
for j in x:
    ax.axvline(j,color='k',linestyle=':')
ax.set_ylim([-2,9])
ax.set_xlim([-1,6])
ax.legend()

ax.draw()