如何在 Python 中使用 matplotlib 在单个页面上绘制多个图?


使用 Pandas,我们可以创建一个数据框架并创建一个图表和轴。之后,我们可以使用 scatter 方法绘制点。

步骤

  • 创建学生列表、他们获得的成绩以及每种成绩对应的颜色编码。

  • 使用 Pandas 的 DataFrame,用步骤 1 中的数据创建一个数据框架。

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

  • 使用 plt.xlabel() 方法设置 X 轴标签。

  • 使用 plt.ylabel() 方法设置 Y 轴标签。

  • 绘制散点图,其中 *y* 与 *x* 对应,标记大小和/或颜色各不相同。

  • 若要显示图表,请使用 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()

输出

更新于:2021-03-16

323 次阅读

开启你的 事业

通过完成课程获得认证

入门
广告