Lập đồ thị dữ liệu phân loại với gấu trúc và matplotlib

Tôi có một khung dữ liệu với dữ liệu phân loại:

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

Tôi muốn tạo một số biểu đồ, như biểu đồ hình tròn và biểu đồ dựa trên các danh mục. Có thể nào mà không tạo biến số giả không? Cái gì đó như

df.plot(kind='hist')

Bạn chỉ có thể sử dụng value_countstrên loạt:

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

Bạn có thể tìm thấy mosaicâm mưu hữu ích từ các mô hình thống kê. Điều này cũng có thể làm nổi bật thống kê cho các phương sai.

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

Nhưng hãy cẩn thận với ô có kích thước 0 - chúng sẽ gây ra các vấn đề với nhãn.

Xem câu trả lời này để biết chi tiết

như thế này :

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

Bạn cũng có thể sử dụng countplottừ seaborn. Gói này được xây dựng dựa trên pandasđể tạo ra một giao diện vẽ đồ thị cấp cao. Nó cung cấp cho bạn kiểu dáng đẹp và nhãn trục chính xác miễn phí.

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')

Nó cũng hỗ trợ tô màu các thanh đúng màu với một mẹo nhỏ

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

Để vẽ nhiều đối tượng địa lý phân loại dưới dạng biểu đồ thanh trên cùng một biểu đồ, tôi sẽ đề xuất:

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()