用 Python 的 pandas 创建 matplotlib 散点图
使用 Pandas,我们可以创建一个数据框,并使用 subplot() 方法创建一个图表和轴变量。随后,我们可以使用 ax.scatter() 方法获取所需的图表。
步骤
制作一份学生人数列表。
制作一份学生获得的成绩列表。
为了表示每个散点的颜色,我们可以制作一份颜色列表。
使用 Pandas,我们可以制作一份表示数据框轴的列表。
使用子区段方法创建图表和轴变量,其中默认的 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()
输出
广告