Сохранить pandas корреляционную матрицу вместо ее отображения

Question

Я хочу сохранить корреляционную матрицу pandas вместо ее отображения. Вот код, который я пробовал. Этот код отображает изображение и сохраняет его. Я просто хочу сохранить график корреляции, но не хочу отображать его.

for subject in range(len(train)):
  for trial in range(subject):
    df = pd.DataFrame(train[subject][trial], columns=channels)
    img=plt.matshow(df.corr())
    plt.savefig('train_images/result_{}_{}.png'.format(subject,trial), bbox_inches='tight', pad_inches=0.0)

Спасибо!

Answer 1

Полагаю, у вас есть этот код в блокноте Jupyter с %matplotlib inline. Затем вы можете просто поставить

plt.close()

после plt.savefig(...), чтобы предотвратить вставку фигуры в блокнот.