Python PandasでDataFrameの複数のデータ列をプロットする方法

PythonのPandasライブラリを使えば、DataFrameに含まれる複数のデータ列を簡単に可視化できます。本記事では、plot()メソッドを使用し、kindパラメータをbarに設定することで、複数の列を1つの棒グラフとして描画する方法を解説します。

必要なライブラリのインポート

まず、PandasとMatplotlibという2つの必須ライブラリをインポートします。Pandasはデータ操作を、Matplotlibはグラフの表示を担当します。

import pandas as pd
import matplotlib.pyplot as mp

サンプルデータの準備

ここでは、各チームのランキングポイントと年を含むチーム記録データを例に使用します。

data = [["Australia", 2500, 2021],["Bangladesh", 1000, 2021],["England", 2000, 2021],["India", 3000, 2021],["Srilanka", 1500, 2021]]

DataFrameの作成

次に、このデータをPandasのDataFrameに変換し、「Team」「Rank_Points」「Year」という3つの列名を設定します。

dataFrame = pd.DataFrame(data, columns=["Team","Rank_Points", "Year"])

複数の列を棒グラフでプロットする

棒グラフとして複数の列をプロットするには、plot()メソッドのkindパラメータに「bar」を指定します。x引数にはX軸に表示する列（ここでは「Team」）を、y引数にはプロットしたい複数の列をリスト形式で渡します。また、figsizeでグラフのサイズを調整できます。

dataFrame.plot(x="Team", y=["Rank_Points","Year" ], kind="bar", figsize=(10, 9))

完全なコード例

以下に、これまでの手順をまとめた完全なコードを示します。

import pandas as pd
import matplotlib.pyplot as mp

# サンプルデータ
data = [["Australia", 2500, 2021],["Bangladesh", 1000, 2021],["England", 2000, 2021],["India", 3000, 2021],["Srilanka", 1500, 2021]]

# DataFrameの作成
dataFrame = pd.DataFrame(data, columns=["Team","Rank_Points", "Year"])

# 複数の列を棒グラフとしてプロット
dataFrame.plot(x="Team", y=["Rank_Points","Year" ], kind="bar", figsize=(10, 9))

# グラフの表示
mp.show()

実行結果

上記のコードを実行すると、X軸にチーム名が表示され、「Rank_Points」と「Year」の2つの列が並んだ棒グラフが出力されます。このように、y引数にリストで複数の列名を渡すだけで、1つのグラフ上に複数のデータ系列を簡単に重ねて表示できるのがPandasの強みです。