python - 在散点图上用不同颜色分隔不同的点

Question

这是我编写的代码:

import pandas as pd
import matplotlib.pyplot as plt

data1 = pd.read_csv('F:\HCSE\sample_data1.csv',sep=';')
colnames = data1.columns

plt.plot(data1.iloc[:,0],data1.iloc[:,2],'bs')
plt.ylabel(colnames[2])
plt.xlabel(colnames[0])
plt.show()

这是我使用过的数据:

Age;Gender;LOS;WBC;HB;Nothrophil
0.62;1;0.11;9.42;22.44;70.43
0.84;0;0.37;4.4;10.4;88.4
0.78;0;0.23;6.8;15.6;76.5
0.8;0;-0.02;9.3;15.1;87
0.7;1;0.19;5.3;11.3;82
0.25;0;0.27;5.9;10.6;87.59
0.32;0;0.37;3.1;12.5;15.4
0.86;1;0.31;4.1;10.4;77
0.75;0;0.21;12.07;14.1;88

最后，我画了一个图表，可以在 link here. 中找到。

我的问题是:如何为不同的性别设置不同的颜色(例如:male=red 和female=blue)？

提前致谢

Answer 1

我认为您正在寻找这样的东西:

cols = {0: 'red', 1: 'blue'}

plt.scatter(data1.Age, data1.LOS, c=data1.Gender.map(cols))