Построение двух тепловых карт в Matplotlib - PullRequest
Новая
       22

Построение двух тепловых карт в Matplotlib

3 голосов
/ 13 апреля 2019

У меня есть функция, которая отображает тепловую карту для корреляционной матрицы DataFrame.Функция выглядит так: 

def corr_heatmap(data):
    columns = data.columns
    corr_matrix = data.corr()

    fig, ax = plt.subplots(figsize=(7, 7))
    mat = ax.matshow(corr_matrix, cmap='coolwarm')

    ax.set_xticks(range(len(columns)))
    ax.set_yticks(range(len(columns)))
    ax.set_xticklabels(columns)
    ax.set_yticklabels(columns)
    plt.setp(ax.get_xticklabels(), rotation=45, ha='left', rotation_mode='anchor')
    plt.colorbar(mat, fraction=0.045, pad=0.05)
    fig.tight_layout()
    plt.show()

    return mat

и при запуске с DataFrame выдает что-то вроде этого:

enter image description here

Что я хочусделать, это построить две из этих тепловых карт бок о бок, но у меня возникли некоторые проблемы с этим.До сих пор я пытался назначить каждую тепловую карту объекту AxesImage и использовать вспомогательные участки для их построения. 

mat1 = corr_heatmap(corr_mat1)
mat2 = corr_heatmap(corr_mat2)

fig = plt.figure(figsize=(15, 15))
ax1 = fig.add_subplot(221)
ax2 = fig.add_subplot(222)
ax1.plot(ma1)
ax2.plot(ma2)

, но это дает мне следующую ошибку: 

TypeError: float() argument must be a string or a number, not 'AxesImage'

Может быть, кто-нибудь знает, как я могу построить два изображения тепловых карт рядом?Спасибо.

РЕДАКТИРОВАТЬ

В случае, если кому-то интересно, как будет выглядеть окончательный код для того, что я хотел сделать: 

def corr_heatmaps(data1, data2, method='pearson'):

    # Basic Configuration
    fig, axes = plt.subplots(ncols=2, figsize=(12, 12))
    ax1, ax2 = axes
    corr_matrix1 = data1.corr(method=method)
    corr_matrix2 = data2.corr(method=method)
    columns1 = corr_matrix1.columns
    columns2 = corr_matrix2.columns

    # Heat maps.
    im1 = ax1.matshow(corr_matrix1, cmap='coolwarm')
    im2 = ax2.matshow(corr_matrix2, cmap='coolwarm')

    # Formatting for heat map 1.
    ax1.set_xticks(range(len(columns1)))
    ax1.set_yticks(range(len(columns1)))
    ax1.set_xticklabels(columns1)
    ax1.set_yticklabels(columns1)
    ax1.set_title(data1.name, y=-0.1)
    plt.setp(ax1.get_xticklabels(), rotation=45, ha='left', rotation_mode='anchor')
    plt.colorbar(im1, fraction=0.045, pad=0.05, ax=ax1)

    # Formatting for heat map 2.
    ax2.set_xticks(range(len(columns2)))
    ax2.set_yticks(range(len(columns2)))
    ax2.set_xticklabels(columns2)
    ax2.set_yticklabels(columns2)
    ax2.set_title(data2.name, y=-0.1)
    plt.setp(ax2.get_xticklabels(), rotation=45, ha='left', rotation_mode='anchor')
    plt.colorbar(im2, fraction=0.045, pad=0.05, ax=ax2)

    fig.tight_layout()

Thisмог (при запуске с двумя Pandas DataFrames) выводить что-то по следующему изображению:

enter image description here

Ответы [ 2 ]

2 голосов
/ 13 апреля 2019

Вам нужна функция plt.subplots.Вместо ручного добавления Axes объектов к Figure, вы можете инициализировать Figure вместе с числом Axes.Тогда это так же просто, как вызов matshow для каждого Axes: 

import numpy as np
import pandas as pd

from matplotlib import pyplot as plt

df = pd.DataFrame(np.random.rand(10, 10))

fig, axes = plt.subplots(ncols=2, figsize=(8, 4))

ax1, ax2 = axes

im1 = ax1.matshow(df.corr())
im2 = ax2.matshow(df.corr())

fig.colorbar(im1, ax=ax1)
fig.colorbar(im2, ax=ax2)

enter image description here

Вы можете выполнить все остальные форматирования позже.

0 голосов
/ 13 апреля 2019

Пожалуйста, следуйте приведенному ниже примеру, измените график на matshow, выполните настройку оси в соответствии с вашими потребностями. 

import numpy as np 
import matplotlib.pyplot as plt 

def f(t): 
    return np.exp(-t) * np.cos(2*np.pi*t) 

t1 = np.arange(0.0, 3.0, 0.01) 

ax1 = plt.subplot(121) 
ax1.plot(t1, f(t1), 'k') 

ax2 = plt.subplot(122) 
ax2.plot(t1, f(t1), 'r') 
plt.show()

Выход:

Output

...