Сиборн: построение двух парных участков рядом - PullRequest
Новая
       63

Сиборн: построение двух парных участков рядом

1 голос
/ 08 мая 2020

Я пытаюсь создать функцию, которая создает две пары рядом друг с другом. Я должен произвести 1 раз. 

import seaborn as sns
data = sns.load_dataset("iris")

def pairplot(col,hue_var):
    sns.set(style="ticks")
    cols_to_look_at = col + [hue_var]
    sns.pairplot(data[cols_to_look_at], hue=hue_var)

col_names1 =  ['sepal_length', 'sepal_width']
col_names2 =  ['petal_length', 'petal_width']

pairplot(col_names1,'species')
pairplot(col_names2,'species')

благодарен за любую помощь

1 Ответ

1 голос
/ 09 мая 2020

На самом деле это нетривиальный вопрос.

pairplot не имеет параметра ax, чтобы упростить его из-за его природы "уровня фигуры" на море.

Есть несколько приемов: Как построить несколько совместных сюжетов Моря на подзаголовке

«Чистый» состоит в сохранении графиков в виде изображений, чтобы вернуть их, чтобы разместить их: { ссылка }

Использование на своих участках: 

import matplotlib.image as mpimg
import matplotlib.pyplot as plt


import seaborn as sns
data = sns.load_dataset("iris")


def pairplot(col,hue_var):

    sns.set(style="ticks")
    cols_to_look_at = col + [hue_var] 
    return sns.pairplot(data[cols_to_look_at], hue=hue_var) # !!! I've placed a return

col_names1 =  ['sepal_length', 'sepal_width']
col_names2 =  ['petal_length', 'petal_width']

g0 = pairplot(col_names1,'species')
g1 = pairplot(col_names2,'species')



############### 1. SAVE PLOTS IN MEMORY TEMPORALLY
g0.savefig('g0.png', dpi=300)
plt.close(g0.fig)

g1.savefig('g1.png', dpi=300)
plt.close(g1.fig)


############### 2. CREATE YOUR SUBPLOTS FROM TEMPORAL IMAGES
f, axarr = plt.subplots(1, 2, figsize=(20, 20))

axarr[0].imshow(mpimg.imread('g0.png'))
axarr[1].imshow(mpimg.imread('g1.png'))

# turn off x and y axis
[ax.set_axis_off() for ax in axarr.ravel()]

plt.tight_layout()
plt.show()
...