sns.headmap (df.corr (), annot = True, cmap = cmap, ax = ax) в цикле for - PullRequest
Новая
       9

sns.headmap (df.corr (), annot = True, cmap = cmap, ax = ax) в цикле for

0 голосов
/ 29 октября 2018

У меня есть список файлов pd.df, для которых я хочу сохранить тепловую карту их корреляции (df.corr ()) отдельно в виде файла .jpeg. Я добавил небольшое форматирование в тепловую карту, используя cmap и figsize. 

import mathplotlib.pyplot as plt
import seaborn as sns
import pandas as pd

%mathplotlib inline

cmap=sns.diverging_palette(250, 50,as_cmap=True)
_,ax = plt.subplots(figsize=(10,7))

df_list = [df1,df2,..dfn]
corr_list = []
for df in df_list:
    corr = sns.headmap(df.corr(),annot = True,cmap=cmap,ax=ax)
    corr_list.append(corr)
    plt.show()

Я ожидал, что данные тепловых карт будут выводиться как результат, но результат будет 

matplotlib.axes._subplots.AxesSubplot
matplotlib.axes._subplots.AxesSubplot
matplotlib.axes._subplots.AxesSubplot
(... ntimes)

Даже plt.savefig() сохраняет пустую фигуру. При попытке итерации corr_list я заканчиваю тем же.

Что мне не хватает?

...