共享轴并删除 matplotlib 子图中未使用的

Question

我想在网格中绘制一系列 seaborn 热图。我知道子图的数量（可以是奇数或偶数）。热图将通过 "day of week"（y 轴）和 "hour of day"（x 轴）显示平均值 "occupation ratio"，例如他们都共享相同的 x / y 域。

这是我当前的代码：

df2 = df[['name','openLots','occupationRatio','DoW','Hour']]
fig, axs = plt.subplots(figsize=(24,24), nrows=7, ncols=6)
axs = axs.flatten()
locations = df2['name'].sort_values().unique()


def occupation_heatmap (name, ax):
    dfn = df2[df2['name'] == name]
    dfn = dfn.groupby(['DoW', 'Hour']).mean()['occupationRatio'].unstack()
    dfn = dfn.reindex(['Mon', 'Tue', 'Wed','Thu','Fri','Sat','Sun'])
    sns.heatmap(data=dfn, cmap="coolwarm", vmin=0, vmax=1.0, ax= ax)
    ax.set_title(name)


i = 0
for n in locations: 
    occupation_heatmap (n, axs[i])
    i = i+1

plt.tight_layout()

看起来几乎像我想要的（最后几行）：不管我想要什么：

每行只有一次 y 轴标签 (DoW)（最左边的图）
仅在每行最右边的图上使用颜色图图例（或完全不使用，颜色非常不言自明）
去掉最后一行的"empty plots"，因为总数是奇数

非常感谢任何提示

Answer 1

您可以更加灵活，只需为每个出现的名称创建一个轴，如下所示：

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
import seaborn as sns
import string

days = ['Mon','Tue','Wed','Thurs','Fri','Sat','Sun']
names = [string.lowercase[i] for i in range(22)]

nItems = 1000

df = pd.DataFrame()
df['name'] = [names[i] for i in np.random.randint(0,len(names),nItems)]
df['openLots'] = np.random.randint(0,100,nItems)
df['occupationRatio'] = np.random.randint(0,100,nItems)
df['DoW'] = [days[i] for i in np.random.randint(0,7,nItems)]
df['Hour'] = np.random.randint(0,12,nItems)




fig = plt.figure(figsize=(12,12))
for index, name in enumerate(names):
    ax = fig.add_subplot(4,6,index+1)
    dfn = df.loc[df.name==name]
    dfn = dfn.groupby(['DoW','Hour']).mean()['occupationRatio'].unstack()
    dfn = dfn.reindex(days)

    # Now we can operate on each plot axis individually
    if index%6!=5: #i.e.
        # Don't draw a colorbar
        sns.heatmap(data = dfn, cmap='coolwarm', ax=ax, cbar=False)
    else:
        sns.heatmap(data = dfn, cmap='coolwarm', ax=ax)

    if index%6!=0:
        # Remove the y-axis label
        ax.set_ylabel('')
        ax.set_yticks(())

    ax.set_title(name)

fig.tight_layout()
fig.show()

结果：您还可以使用 x 轴（例如，删除除底行之外的标签和刻度）。

Answer 2

每行只有一次 y 轴标签 (DoW)（最左边的图）
这可以使用 sharey = True 作为 plt.subplots.
仅在每行最右边的图上使用颜色图图例（或者完全不使用，颜色非常不言自明）
使用 cbar = False 参数到 seaborn.heatmap 以便不显示颜色条。这可以作为绘图函数的输入，具体取决于子图的实际数量。
去掉最后一行的"empty plots"因为总数是奇数
在创建绘图的循环之后，您可以添加另一个循环来删除未使用的轴。
```
for j in range(len(locations), ncols*nrows):
    axs[j].axis("off")
```

这是一个完整的例子（我借用了@Robbie 的 cod 来生成数据框）：

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
import seaborn as sns

days = ['Mon','Tue','Wed','Thurs','Fri','Sat','Sun']
names = ["Parkhaus {:02}".format(i+1) for i in range(22)]

nItems = 1000

df = pd.DataFrame()
df['name'] = [names[i] for i in np.random.randint(0,len(names),nItems)]
df['openLots'] = np.random.randint(0,100,nItems)
df['occupationRatio'] = np.random.rand(nItems)
df['DoW'] = [days[i] for i in np.random.randint(0,7,nItems)]
df['Hour'] = np.random.randint(0,12,nItems)

df2 = df[['name','openLots','occupationRatio','DoW','Hour']]
nrows = 4; ncols=6
fig, axs = plt.subplots(nrows=nrows, ncols=ncols, figsize=(15,9), sharey=True)
axs = axs.flatten()
locations = df2['name'].sort_values().unique()


def occupation_heatmap (name, ax, cbar=False, ylabel=False):
    dfn = df2[df2['name'] == name]
    dfn = dfn.groupby(['DoW', 'Hour']).mean()['occupationRatio'].unstack()
    dfn = dfn.reindex(['Mon', 'Tue', 'Wed','Thu','Fri','Sat','Sun'])
    sns.heatmap(data=dfn, cmap="coolwarm", vmin=0, vmax=1.0, ax=ax, cbar=cbar)
    ax.set_title(name)
    plt.setp(ax.get_yticklabels(), rotation=0)
    if not ylabel: ax.set_ylabel("")


for i, n in enumerate(locations): 
    occupation_heatmap (n, axs[i], cbar=i%ncols==ncols-1, ylabel=i%ncols==0)
for j in range(len(locations), ncols*nrows):
    axs[j].axis("off")

plt.tight_layout()
plt.show()

共享轴并删除 matplotlib 子图中未使用的

Share axis and remove unused in matplotlib subplots

python

matplotlib

pandas

seaborn