python绘图之台阶图方案

定义函数

def draw_stage(df, colour):
    #调整画布尺寸
    plt.figure(figsize=(5, 3), dpi = 600)#dpi调整清晰度
    plt.ylim((-0.5, 2.0))
    plt.xticks([])
    #y轴的显示范围
    plt.yticks(np.arange(-0.5, 2.01, 0.5), fontproperties = 'Arial')

    #通过第一个虚线把虚线画出来
    for col in df.columns[1:]:
        plt.plot(df['x'], df[col], c='#a0d8ef', ls='--', lw=1)
    #通过第二个虚线把短横线画出来
    a_1 = df['x'].iloc[0::2].values
    a_2 = df['x'].iloc[1::2].values
    for col_1, c_1 in zip(df.columns[1:], colour):
        a_3 = df[col_1].iloc[0::2].values
        a_4 = df[col_1].iloc[1::2].values
        x_1 = np.array([a_1, a_2])
        y_1 = np.array([a_3, a_4])
        plt.plot(x_1, y_1, c=c_1)
    #选择保存位置    
    plt.savefig('D:\Myjob\Pc\stage.png')
    plt.show()

传递给函数值

df = pd.read_excel(r'D:\Myjob\Pc\python_stage_U.xlsx')
#绘图的文件就放在这个位置
colour = ['r', 'g']
draw_stage(df, colour)