使用 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()
输出
广告