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' 카테고리에 할당합니다. , 'B' 및 'C'입니다.

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 중국어 웹사이트의 기타 관련 기사를 참조하세요!