颜色对应字符串的Python散点图

Question

我想用 python matplotlib 绘制一个散点图，其中点的颜色应该与数据文件中的特定字符串相对应，所以像这样:

data = np.genfromtxt('filename.txt', delimiter=',', dtype=None, names=['a', 'b', 'c'])
plt.scatter(data['a'], data['b'])

其中文件“a”的第一列是 float ，第二列“b”是 float ，第三列“c”是字符串。字符串列包含 5 个不同的词，我想将其绘制为 5 种不同的颜色是散点图。有任何想法吗？非常感谢!

Answer 1

按照这些思路应该可以解决问题:

color_dict = { 'Allan':'red', 'Betty':'blue', 'Chris':'black', 'Diane':'green','Eugene':'purple' }

plt.scatter( data['a'], data['b'], color=[ color_dict[i] for i in data['c'] ] )