Python Matplotlib : Splitting one Large Graph into several Sub-Graphs (Subplot)

Question

简单地说，假设我有 2 个列表:
A -> 具有名称列表 ['A','B','C','D','E','F','G','H']
B -> 具有值列表 [5,7,3,8,2,9,1,3]
A 将是 X 轴标签的名称，B 中的相应值将是图形的高度(即 Y 轴)。

%matplotlib inline
import pandas as pd
from matplotlib import rcParams
import matplotlib.pyplot as plt
from operator import itemgetter

rcParams.update({'figure.autolayout': True})
plt.figure(figsize=(14,9), dpi=600)

reso_names = ['A','B','C','D','E','F','G','H']
reso_values = [5,7,3,8,2,9,1,3]
plt.bar(range(len(reso_values)), reso_values, align='center')
plt.xticks(range(len(reso_names)), list(reso_names), rotation='vertical')

plt.margins(0.075)
plt.xlabel('X-Axis')
plt.ylabel('Y-Axis')
plt.title('Graph', {'family' : 'Arial Black',
        'weight' : 'bold',
        'size'   : 22})
plt.show()

此代码提供以下输出:

但是我希望它为每 2 个值创建子图。在这种情况下应该有 4 个子图:

• 第一个图有“A”和“B”
• 第二张图有“C”和“D”
• 第三张图有“E”和“F”
• 第四张图有“G”和“H”

这种拆分应该是动态完成的(不是 4 个不同的循环，它应该根据输入的大小将图分成 2 个单元，如果列表 A 有 10 个值，那么它应该给出 5 个子图)。

我想出了如何将图形分成两半，但我需要使用每个图形 N 的步骤来实现它(本例中的 N 为 2)。
我将图分成 2 个相等子图的代码是:

%matplotlib inline
import pandas as pd
from matplotlib import rcParams
import matplotlib.pyplot as plt
from operator import itemgetter

rcParams.update({'figure.autolayout': True})
plt.figure(figsize=(14,9), dpi=600)

reso_names = ['A','B','C','D','E','F','G','H']
reso_values = [5,7,3,8,2,9,1,3]

fig, axs = plt.subplots(nrows=2, sharey=True, figsize=(14,18), dpi=50)
size = int(len(reso_values))
half = int( size/2 )

fig.suptitle('Graph', 
          **{'family': 'Arial Black', 'size': 22, 'weight': 'bold'})

for ax, start, end in zip(axs, (0, half), (half, size)):
    names, values = list(reso_names[start:end]), reso_values[start:end]
    ax.bar(range(len(values)), values, align='center')
    ax.set_xlabel('X-Axis')
    ax.set_ylabel('Y-Axis')
    ax.set_xticks(range(len(names)))
    ax.set_xticklabels(names, rotation='vertical')
    ax.set_xlim(0, len(names))
fig.subplots_adjust(bottom=0.05, top=0.95)
plt.show()

这给了我:

我只是想让程序根据拆分数N将图动态拆分成子图。

Answer 1

您可以直接将包含 size 元素的列表 values/names 拆分为 size//N + 1具有此代码的 N 元素列表:

N=3
sublists_names = [reso_names[x:x+N] for x in range(0, len(reso_names), N)]
sublists_values = [reso_values[x:x+N] for x in range(0, len(reso_values), N)]

注意，如果N不分size，最后一个子列表的元素会比较少。

然后你只需执行一个 zip 并在不同的图表中绘制每个子列表:

import pandas as pd
from matplotlib import rcParams
import matplotlib.pyplot as plt
from operator import itemgetter

rcParams.update({'figure.autolayout': True})
plt.figure(figsize=(14,9), dpi=600)

reso_names = ['A','B','C','D','E','F','G','H']
reso_values = [5,7,3,8,2,9,1,3]

N=3
sublists_names = [reso_names[x:x+N] for x in range(0, len(reso_names), N)]
sublists_values = [reso_values[x:x+N] for x in range(0, len(reso_values), N)]

size = int(len(reso_values))
fig, axs = plt.subplots(nrows=size//N+1, sharey=True, figsize=(14,18), dpi=50)

fig.suptitle('Graph', 
          **{'family': 'Arial Black', 'size': 22, 'weight': 'bold'})

for ax, names, values in zip(axs, sublists_names, sublists_values):
    ax.bar(range(len(values)), values, align='center')
    ax.set_xlabel('X-Axis')
    ax.set_ylabel('Y-Axis')
    ax.set_xticks(range(len(names)))
    ax.set_xticklabels(names, rotation='vertical')
    ax.set_xlim(0, len(names))
    #ax.set_xlim(0, N)

fig.subplots_adjust(bottom=0.05, top=0.95)
plt.show()

如果列表不能被 N 整除，您可以取消注释最后注释的行，以便条形图在最后一个子图中保持对齐:(ax.set_xlim(0, N)) :