python - 如何使用多维 numpy 数组作为输入用 Seaborn 绘制箱线图或小提琴图？

Question

我有一个尺寸为 (500, 10) 的 2D 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？谢谢。

标签： pythonnumpyseaborn

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