将颜色条添加到pyplot

时间:2019-01-24 19:12:32

标签: python matplotlib

我对matplotlib还是很陌生,但是在给我的图形添加颜色条时遇到了麻烦。

我当前的代码在这里:

import matplotlib.pyplot as plt
import pandas as pd
import matplotlib.cm as cm
import glob
from natsort import natsorted
import re

plt.xlabel('Displacement (A)')
plt.ylabel('Gs(r,t)')
plt.title('Van Hove Self Correlation')

color_num = 0
colors = cm.get_cmap('jet', len(sorted(glob.glob('van_hove_t=*.dat'))))

for filename in natsorted(glob.glob('van_hove_t=*.dat')):
    print(filename)
    df = pd.read_csv(filename, sep='\t', skiprows=1, )
    #print(df)
    color_num+=(1/len(sorted(glob.glob('van_hove_t=*.dat'))))
    plt.plot( 'A^-1', 'count', data=df, color=colors(color_num), linewidth=1)

plt.xlim(0, 150)
plt.show()
plt.close()

它获取几个文件的输出,并按顺序绘制它们,并在每个连续的绘制中更改颜色。这是输出示例: enter image description here

我一生都无法弄清楚如何在此绘图中添加颜色条。我尝试过plt.colorbar(),但是我不知道要指向哪个对象来创建颜色条,colorsplt无效。

0 个答案:

没有答案