Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Python Pandas - Nhóm swarm theo hai biến phân loại với Seaborn

Swarm Plot trong Seaborn dùng để vẽ biểu đồ phân tán phân loại với các điểm không chồng chéo. Hàm seaborn.swarmplot() được dùng cho mục này. Để nhóm swarm bằng hai biến phân loại, đặt các biến đó trong swarmplot() sử dụng tham số x, y hoặc hue.

Giả sử tập dữ liệu của chúng ta ở dạng file CSV: Cricketers2.csv

Trước tiên, nhập các thư viện cần thiết:

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

Tải dữ liệu từ file CSV vào Pandas DataFrame:

dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\Cricketers2.csv")

Nhóm swarm bằng hai biến phân loại:

sb.swarmplot(x="Role", y="Matches", hue="Academy", data=dataFrame)

Ví dụ hoàn chỉnh

Mã nguồn đầy đủ:

import seaborn as sb
import pandas as pd
import matplotlib.pyplot as plt
# Tải dữ liệu từ file CSV vào Pandas DataFrame:
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\Cricketers2.csv")
# đặt theme
sb.set_theme(style="whitegrid")
sb.swarmplot(x="Role", y="Matches", hue="Academy", data=dataFrame)
# hiển thị
plt.show()

Kết quả

Chương trình sẽ tạo ra biểu đồ sau:

Python Pandas - Nhóm swarm theo hai biến phân loại với Seaborn