如何使用多维 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])