我对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()
它获取几个文件的输出,并按顺序绘制它们,并在每个连续的绘制中更改颜色。这是输出示例:
我一生都无法弄清楚如何在此绘图中添加颜色条。我尝试过plt.colorbar()
,但是我不知道要指向哪个对象来创建颜色条,colors
和plt
无效。