python - 如何使用 pyplot 绘制二进制值

Question

我正在尝试绘制通过两次考试的 100 名学生的分数。 X 值是第一次考试分数，Y 值是第二次考试分数。我的数据框中有第三列为 0 或 1。如果他们以两次考试的成绩被大学录取，则在我的第三列中用 1 表示。如果他们没有被录取，则为 0。

我试图在他们没有被录取时绘制点，在他们被录取时绘制“+”。

我该怎么做？

这是我现在的代码:

data.plot(kind='scatter', x='First Exam Score', y='Second Exam Score', figsize=(12,8))

这是我现在的图表:

我希望有些点是点，有些是“+”，这取决于它们在第三列中的值

Answer 1

您可以将所选数据绘制到同一轴上:

data = pd.DataFrame({'First': np.random.randint(30,100, 100),
                     'Second': np.random.randint(30,100,100),
                     'Admitted': np.random.randint(0,2,100)})

fig, ax = plt.subplots()
markers = ['x','o']

for i in range(2):
    data[data['Admitted'].eq(i)].plot.scatter(x='First', 
                                              y='Second', 
                                              marker=markers[i],
                                              ax=ax)

输出:

或者你可以使用 seaborn scatterplot 和 style:

但在我看来，更好的方法是使用 seaborn 进行颜色编码:

sns.scatterplot(data=data, x='First', y='Second', hue='Admitted', style='Admitted')

输出: