การพล็อตข้อมูลเชิงหมวดหมู่ด้วยแพนด้าและ matplotlib

ฉันมีกรอบข้อมูลที่มีข้อมูลหมวดหมู่:

     colour  direction
1    red     up
2    blue    up
3    green   down
4    red     left
5    red     right
6    yellow  down
7    blue    down

ฉันต้องการสร้างกราฟบางอย่างเช่นแผนภูมิวงกลมและฮิสโตแกรมตามหมวดหมู่ เป็นไปได้หรือไม่โดยไม่ต้องสร้างตัวแปรตัวเลขจำลอง สิ่งที่ต้องการ

df.plot(kind='hist')

คุณสามารถใช้value_countsกับซีรีส์:

df['colour'].value_counts().plot(kind='bar')

คุณอาจพบmosaicพล็อตที่เป็นประโยชน์จาก statsmodels ซึ่งสามารถให้การเน้นทางสถิติสำหรับความแปรปรวน

from statsmodels.graphics.mosaicplot import mosaic
plt.rcParams['font.size'] = 16.0
mosaic(df, ['direction', 'colour']);

แต่ระวังเซลล์ขนาด 0 เพราะจะทำให้เกิดปัญหากับป้ายกำกับ

ดูรายละเอียดคำตอบนี้

แบบนี้ :

df.groupby('colour').size().plot(kind='bar')

คุณยังสามารถใช้countplotจากseaborn. แพคเกจนี้สร้างขึ้นpandasเพื่อสร้างอินเทอร์เฟซการลงจุดระดับสูง ให้สไตล์ที่ดีและป้ายแกนที่ถูกต้องฟรี

import pandas as pd
import seaborn as sns
sns.set()

df = pd.DataFrame({'colour': ['red', 'blue', 'green', 'red', 'red', 'yellow', 'blue'],
                   'direction': ['up', 'up', 'down', 'left', 'right', 'down', 'down']})
sns.countplot(df['colour'], color='gray')

นอกจากนี้ยังสนับสนุนการระบายสีแถบด้วยสีที่เหมาะสมด้วยเคล็ดลับเล็กน้อย

sns.countplot(df['colour'],
              palette={color: color for color in df['colour'].unique()})

ในการพล็อตคุณลักษณะเชิงหมวดหมู่หลายรายการเป็นแผนภูมิแท่งในพล็อตเดียวกันฉันขอแนะนำ:

import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame(
    {
        "colour": ["red", "blue", "green", "red", "red", "yellow", "blue"],
        "direction": ["up", "up", "down", "left", "right", "down", "down"],
    }
)

categorical_features = ["colour", "direction"]
fig, ax = plt.subplots(1, len(categorical_features))
for i, categorical_feature in enumerate(df[categorical_features]):
    df[categorical_feature].value_counts().plot("bar", ax=ax[i]).set_title(categorical_feature)
fig.show()