Python散点图，颜色对应于字符串

Question

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

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

其中文件“a”的第一列是浮点数，第二列“b”是浮点数，第三列“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'] ] )