来自pandas DF的matplotlib小提琴剧情

Question

我有一个包含两列的pandas数据框，ArrDelay（一个整数值）和Month（1到12之间的整数表示月份）。

我正在尝试用每个月单独的“小提琴”创作一个小提琴情节。

但是，使用以下代码：

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

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

给我一​​个带有两个“小提琴”的情节，一个用于ArrDelay，一个用于月（我相信）。

如何创建所需的情节？

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