如何創建作爲繪製matplotlib.pyplot.step（X，Y）的步驟（相當於Matlab的樓梯）

Question

有兩個numpy的數組numpy的數組：

>>> x = np.array([0,1,2,3]) 
>>> y = np.array([-1,1,-1,1]) 

有沒有來創建等同陣列XB功能，yb，類似於Matlab的功能[xb,yb]=stairs(x,y)？其中：

>>> xb 
array([0,0,1,1,2,2,3]) 
>>> yb 
array([-1,1,1,-1,-1,1,1]) 

或者也許從matplotlib.pyplot.step（x，y）檢索數組？

Answer 1

一個簡單的配方是使用numpy.repeat

xb = numpy.repeat(x, 2)[:-1] 
yb = numpy.repeat(y, 2)[1:]