如何使用多維 numpy 數組作為輸入使用 Seaborn 繪制箱線圖或小提琴圖？

Question

我有一個尺寸為 (500, 10) 的二維 numpy 數組，我想將其繪制為 Seaborn violinplot 或 boxplot，其中 10 列中的每一列都有一個框。 將它傳遞給 Seaborn 的最簡潔方法是什么，而無需先進行一堆繁瑣的操作以將其放入 Pandas Dataframe 中？ 我相信我可以進行轉換，但恐怕我可能不了解這樣做的最佳和最簡潔的方法。

例如，我可以做類似的事情

all_arrays = []
cols = the_array.shape[1]
for col in range(0, cols):
     all_arrays.append(the_array[:, col])
sns.boxplot(data=all_arrays)

但是有沒有更好的方法將原始數組拆分為數組列表，或者有更好的方法將其傳遞給 seaborn？ 謝謝。

Answer 1

您的解決方案是正確的， boxplot()需要一個向量列表，因此您必須以某種方式將矩陣轉換為該列表。

但是，您可以簡化編寫代碼的方式： sns.boxplot(data=[d for d in the_array.T])

完整代碼：

# create a dummy matrix 500x10
the_array = np.zeros(shape=(500,10))
for i in range(10):
    the_array[:,i] = np.random.normal(loc=i, size=(500,))

sns.boxplot(data=[d for d in the_array.T])