來自熊貓DF的matplotlib小提琴圖

Question

我有一個熊貓數據框，其中有兩列ArrDelay（一個整數值）和Month（介於1到12之間的整數，代表月份）。

我正在嘗試用每個月的單獨“小提琴”創建一個小提琴圖。

但是，使用以下代碼：

flights_subset = flights[["ArrDelay","Month"]].values

fig, axes = plt.subplots()
axes.violinplot(flights_subset)
plt.show()

給我一個情節，上面有兩把“小提琴”，一個是ArrDelay，一個是Month（我相信）。

如何創建所需的圖？

Answer 1

我使用以下代碼解決了這個問題：

flights_subset = []

for x in np.unique(flights[["Month"]].values):
    flights_subset.append(flights[flights["Month"] == x]["ArrDelay"].values)

fig, axes = plt.subplots()
axes.violinplot(flights_subset, positions=np.unique(flights[["Month"]].values))
plt.show()