Matplotlib で散布図のカテゴリを色分けする方法

Matplotlib で異なるカテゴリレベルに異なる色をプロットする方法

問題

カテゴリ変数を含む DataFrame が与えられた場合、散布図を作成したいとしますここで、各カテゴリには独自の色があります。

Matplotlib を使用した解決策

Matplotlib でさまざまなカテゴリの色を指定するには、plt.scatter の c 引数を使用します。この引数は、色の配列、またはカテゴリを色にマッピングするマッピングを受け入れます。

例を次に示します:

<code class="python">import matplotlib.pyplot as plt
import pandas as pd

# Define a DataFrame
df = pd.DataFrame({'category': ['A', 'B', 'C'], 'value': [10, 20, 30]})

# Create the scatter plot
colors = {'A': 'red', 'B': 'green', 'C': 'blue'}
plt.scatter(df['category'], df['value'], c=df['category'].map(colors))
plt.show()</code>

このコードは、赤、緑、青の色をカテゴリ 'A' に割り当てます。

DataFrame GroupBy と Plotting

または、DataFrame.groupby() と .plot() を使用して同じ結果を得ることができます。

<code class="python">fig, ax = plt.subplots(figsize=(6, 6))

df.groupby('category').plot(ax=ax, kind='scatter', x='category', y='value', color=colors)
plt.show()</code>

このコードは、カテゴリを色にマッピングする色辞書の存在を前提としています。

以上がMatplotlib で散布図のカテゴリを色分けする方法の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。