Pythonのパンダのデータフレームからmatplotlib散布図を作成する

Pandasを使用すると、データフレームを作成でき、subplot（）メソッドを使用してFigureとAxes変数を作成できます。その後、ax.scatter（）メソッドを使用して、必要なプロットを取得できます。

ステップ

• 生徒数のリストを作成します。

• 生徒が取得したマークのリストを作成します。

• 各散乱点の色を表すために、色のリストを作成できます。

• パンダを使用すると、データフレームの軸を表すリストを作成できます。

• subplotsメソッドを使用してfig変数とax変数を作成します。デフォルトのnrowsとncolsは1です。

• plt.xlabel（）メソッドを使用して「Studentscount」ラベルを設定します。

• plt.ylabel（）メソッドを使用して「取得マーク」ラベルを設定します。

• 散布点を作成するには、手順4で作成したデータフレームを使用します。点はstudents_count、marks、colorです。

• 図を表示するには、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()

出力