Python: динамически создавать графики морского побережья, а затем отображать результаты бок о бок? - PullRequest
0 голосов
/ 02 июля 2018

Я пытаюсь создать несколько seaborn графиков плотности ядра для числовых переменных моего Pandas DataFrame. У меня есть имена всех моих числовых столбцов в списке, numberCol. В настоящее время я могу сделать kdeplot для каждой переменной, которую я явно назвал, например:

import seaborn as sbn
sbn.set_style('whitegrid')
sbn.kdeplot(np.array(df.v2), bw=0.5) # for pandas.core.frame.DataFrame input

Есть ли лучший способ перебрать список numberCol, создать sbn.kdeplot для каждой переменной в numberCol, а затем отобразить их рядом с чем-то более умным, чем что-то вроде:

import matplotlib.pyplot as plt
import seaborn as sns

# Here we create a figure instance, and two subplots
fig = plt.figure(figsize = (20,20)) # width x height
ax1 = fig.add_subplot(3, 3, 1) # row, column, position
ax2 = fig.add_subplot(3, 3, 2)
ax3 = fig.add_subplot(3, 3, 3)

# We use ax parameter to tell seaborn which subplot to use for this plot
sns.heatmap(data=subset1.corr(), ax=ax1, cmap = cmap, square=True, cbar_kws={'shrink': .3}, annot=True, annot_kws={'fontsize': 12})
sns.heatmap(data=subset2.corr(), ax=ax2, cmap = cmap, square=True, cbar_kws={'shrink': .3}, annot=True, annot_kws={'fontsize': 12})
sns.heatmap(data=subset3.corr(), ax=ax3, cmap = cmap, square=True, cbar_kws={'shrink': .3}, annot=True, annot_kws={'fontsize': 12})

1 Ответ

0 голосов
/ 02 июля 2018

Если я понимаю ваш вопрос, это должно сработать

Ncols = 9
cols = ['col_{:d}'.format(i) for i in range(Ncols)]
df = pd.DataFrame(np.random.random(size=(1000,Ncols)),columns=cols)

fig, axs = plt.subplots(3,3) # adjust the geometry based on your number of columns to plot
for ax,col in zip(axs.flatten(), cols):
    sns.kdeplot(df[col], ax=ax)

enter image description here

...