使用 Matplotlib 为不同的类别级别绘制不同的颜色


我们可以绘制一个图表,其中学生人数将绘制在 X 轴上,学生获得的分数将绘制在 Y 轴上。此外,我们可以为学生获得的不同分数设置颜色。

步骤

  • 列出学生人数。

  • 列出学生获得的分数。

  • 为了表示每个散点图点的颜色,我们可以列出颜色。

  • 使用 Panda,我们可以列出一个表示数据帧的 Axes。

  • 使用子图方法创建 fig 和 ax 变量,默认 nrows 和 ncols 为 1。

  • 使用 plt.xlabel() 方法设置“学生数量”标签。

  • 使用 plt.ylabel() 方法设置“获得的分数”标签。

  • 要创建一个散点图点,请使用一个数据帧,即在步骤 4 中创建的点,即学生数量、分数和颜色。

  • 要显示图形,请使用 plt.show() 方法。

示例

from matplotlib import pyplot as plt
import pandas as pd

no_of_students = [1, 2, 3, 5, 7, 8, 9, 10, 30, 50]
marks_obtained_by_student = [100, 95, 91, 90, 89, 76, 55, 10, 3, 19]
color_coding = ['red', 'blue', 'yellow', 'green', 'red', 'blue', 'yellow', 'green', 'yellow', 'green']

df = pd.DataFrame(dict(students_count=no_of_students,
marks=marks_obtained_by_student, color=color_coding))

fig, ax = plt.subplots()

plt.xlabel('Students count')
plt.ylabel('Obtained marks')

ax.scatter(df['students_count'], df['marks'], c=df['color'])

plt.show()

输出

更新于: 15-Mar-2021

1K+ 次浏览

开启你的 职业生涯

完成课程即可获得认证

开始
广告