具有与字符串相对应的颜色的 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'] ] )