我有一个带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]对其进行着色?
答案 0 :(得分:0)
从scatter上的文档中,您应该使用c
关键字参数:
c:颜色,序列或颜色序列,可选,默认:'b' 标记颜色。可能的值:
- 单色格式字符串。
- 长度为n。
的一系列颜色规格- 使用cmap和norm映射到颜色的n个数字序列。
- 二维数组,其中行为RGB或RGBA。
您的案例是第三种选择,即"使用cmap
和norm
"映射到颜色的n个数字序列,它们也是关键字参数。
plt.scatter(tica_features[2][:,0],tica_features[2][:,1], c=a[:,1], cmap="viridis")