Matplotlib / Seaborn 不同数据大小的小提琴图

Question

我有3个一维数据arrays A，B，C。它们的长度都不一样。

我想用 3 把小提琴制作一把小提琴 plot，每个阵列一个。 我该怎么做呢？

编辑：我已经通过编写代理 function 解决了这个问题，但是必须将每个数组的标签转换为列感觉很浪费。 有没有可能做得更好/更有效

def dict2pandas(d, keyname, valname):
    dframes = []
    for k,v in d.items():
        dframes += [pd.DataFrame({keyname : [k] * len(v), valname : v})]
    return pd.concat(dframes)

data = {
    'A' : np.random.normal(1, 1, 100),
    'B' : np.random.normal(2, 1, 110),
    'C' : np.random.normal(3, 1, 120)
}

dataDF = dict2pandas(data, 'arrays', 'values')

fig, ax = plt.subplots()
sns.violinplot(data=dataDF, x='arrays', y='values', scale='width', axis=ax)
plt.show()

Answer 1

虽然它大致相同，但你可以用 nan 填充你的 numpy arrays ，所以它们的大小都是一样的。 然后可以将它们放入 dataframe 中以使用 seaborn 进行绘图：

data = {
    'A' : np.random.normal(1, 1, 100),
    'B' : np.random.normal(2, 1, 110),
    'C' : np.random.normal(3, 1, 120)
}
maxsize = max([a.size for a in data.values()])
data_pad = {k:np.pad(v, pad_width=(0,maxsize-v.size,), mode='constant', constant_values=np.nan) for k,v in data.items()}
df = pd.DataFrame(data_pad)

fig, ax = plt.subplots()
sns.violinplot(data=df)

Answer 2

我也找不到比用NaN填充 Pandas DataFrame 更好的主意，但这种方法可能更简洁：

import numpy as np
import pandas as pd
import seaborn as sns

# OP's data
data = {
    'A' : np.random.normal(1, 1, 100),
    'B' : np.random.normal(2, 1, 110),
    'C' : np.random.normal(3, 1, 120)
}

# Create DataFrame where NaNs fill shorter arrays 
df = pd.DataFrame([data['A'], data['B'], data['C']]).transpose()

# Label the columns of the DataFrame
df = df.set_axis(['A','B','C'], axis=1)

# Violin plot  
sns.violinplot(data=df)