python - 创建子图而不是单独的图

Question

我有以下数据框:

symbol     aaa       bbb      ccc     ddd      eee     fff      ggg
aaa                         
bbb       -0.001                        
ccc        0.348    -0.025                  
ddd       -0.42     -0.075   -0.701             
eee       -0.276     0.004   -0.516   0.661         
fff        0.175    -0.107    0.363  -0.521   -0.356        
ggg        0.469     0.012    0.364  -0.519   -0.306    0.306   

我正在根据上述数据框创建两个单独的图，如下所示:

def plot_bar(self, corr_df):

    dfstacked = corr_df.stack().order()
    dfstacked.plot(kind='bar', rot=60)
    plt.show()

def plot_heatmap(self, corr_df):

    corr_df = corr_df.fillna(value=0)
    plt.pcolormesh(corr_df.values, cmap=plt.cm.Blues)
    plt.yticks(np.arange(0.5, len(corr_df.index), 1), corr_df.index)
    plt.xticks(np.arange(0.5, len(corr_df.columns), 1), corr_df.columns)
    plt.show()

现在我需要做的不是单独绘制这两个图表，而是需要将它们放在一个网格中。我已经成功创建了以下子图，它需要左侧的条形图和右侧的矩阵/颜色网格:

fig, axes = plt.subplots(2)
axes[0] = plt.subplot2grid((1,5), (0,0), colspan=3)
axes[1] = plt.subplot2grid((1,5), (0,3), colspan=2)
plt.show()

在我的一生中，我似乎无法让它发挥作用。我需要 axes[0] 上的条形图和 axes[1]

上的矩阵/颜色网格

Answer 1

你尝试过吗

fig, axes = plt.subplots(2)

plt.subplot2grid((1,5), (0,0), colspan=3)
# here plot something

plt.subplot2grid((1,5), (0,3), colspan=2)
# here plot something

plt.show()

例如

import matplotlib.pyplot as plt

fig, axes = plt.subplots(2)

plt.subplot2grid((1,5), (0,0), colspan=3)
plt.plot([1,2,3]) # plot something

plt.subplot2grid((1,5), (0,3), colspan=2)
plt.plot([1,2,1]) # plot something

plt.show()

<小时/>

编辑:

import pandas as pd
import numpy as np

def plot_bar(corr_df):

    dfstacked = corr_df.stack().order()
    dfstacked.plot(kind='bar', rot=60)

def plot_heatmap(corr_df):

    corr_df = corr_df.fillna(value=0)
    plt.pcolormesh(corr_df.values, cmap=plt.cm.Blues)
    plt.yticks(np.arange(0.5, len(corr_df.index), 1), corr_df.index)
    plt.xticks(np.arange(0.5, len(corr_df.columns), 1), corr_df.columns)


df = pd.DataFrame(range(10))

fig, axes = plt.subplots(2)

plt.subplot2grid((1,5), (0,0), colspan=3)
plot_bar(df)

plt.subplot2grid((1,5), (0,3), colspan=2)
plot_heatmap(df)

plt.show()