【Python】DataFrameの列名に基づいてSeabornのボックスプロットに色を付ける方法

DataFrameの列名に基づいてSeabornのボックスプロットに色を付けることで、複数の列を視覚的に区別しやすくなります。ここでは、Matplotlibと組み合わせて、各列の名前を判定しながらボックスの塗りつぶし色を動的に変更する方法を解説します。

実装の手順

• 図のサイズを設定し、サブプロット間および周囲の余白（パディング）を調整します。
• col1 と col2 の2つの列を持つPandasデータフレームを作成します。
• 水平方向（orient='h'）のボックスプロットを作成します。
• ax.artists からボックスのアーティストを取得します。
• ボックスを反復処理し、列名に応じて set_facecolor() で塗りつぶし色を設定します。
• 図を表示するには、show() メソッドを使用します。

コード例

import seaborn as sns
import matplotlib.pyplot as plt
import pandas as pd

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

df = pd.DataFrame(
    [[2, 4],
    [7, 2]
    ], columns=['col1', 'col2'])

ax = sns.boxplot(data=df, orient='h')
boxes = ax.artists

for i, box in enumerate(boxes):
    if 'col1' in df.columns[i]:
        box.set_facecolor('r')
    else:
        box.set_facecolor('b')

plt.show()

出力結果

このコードを実行すると、水平方向のボックスプロットが表示されます。列名が「col1」であるボックスは赤（r）、それ以外の列は青（b）で塗りつぶされます。df.columns[i] の判定条件を変更すれば、任意の列名パターンに合わせて柔軟に配色をカスタマイズできます。