使用外部文件的python中的Colormap

时间:2018-05-16 17:26:22

标签: python-3.x matplotlib

我有一个带x和y列的外部文件。我想根据Matplotlib中外部文件的y列为我的数据着色。我想跟着

from matplotlib import pyplot as plt
a=np.genfromtxt('rmsd_tonormal3.xvg')

plt.scatter(tica_features[2][:,0],tica_features[2][:,1],a[:,1])

这似乎有效,但如何根据[:,1]对其进行着色?

1 个答案:

答案 0 :(得分:0)

scatter上的文档中,您应该使用c关键字参数:

  

c:颜色,序列或颜色序列,可选,默认:'b'   标记颜色。可能的值:

     
      
  • 单色格式字符串。
  •   
  • 长度为n。
  • 的一系列颜色规格   
  • 使用cmap和norm映射到颜色的n个数字序列。
  •   
  • 二维数组,其中行为RGB或RGBA。
  •   

您的案例是第三种选择,即"使用cmapnorm"映射到颜色的n个数字序列,它们也是关键字参数。

plt.scatter(tica_features[2][:,0],tica_features[2][:,1], c=a[:,1], cmap="viridis")