python - 如何为 matplotlib 列表中的子批处理设置标题

Question

我尝试为数据数组中的子图设置标题。

我期望的是图中的 (00, 01 , 10 和 11) 位置的 Title-1、Title-2 ...等等。

所以我做到了；

import matplotlib.pyplot as plt    

title = [1,2,3,4]

fig, ax = plt.subplots(2, 2, figsize=(6, 8))  

    for i in range(len(ax)): 
        for j in range(len(ax[i])):

            for k in title:
            #    print (k)
                ax[i,j].set_title('Title-' + str(k))

但只获得 Title-4。我该如何解决这个问题？谢谢

Answer 1

一种方法使用展平和枚举:

import matplotlib.pyplot as plt    

title = [1,2,3,4]

fig, ax = plt.subplots(2, 2, figsize=(6, 8))  
flat_ax = ax.flatten()

for n, ax in enumerate(flat_ax):
    ax.set_title(f'Title-{title[n]}')

输出:

另一种选择是将 iter 与 flatten 结合使用:

import matplotlib.pyplot as plt    

title = [1,2,3,4]
ititle = iter(title)

fig, ax = plt.subplots(2, 2, figsize=(6, 8))  
flat_ax = ax.flatten()

for ax in flat_ax:
    ax.set_title(f'Title-{next(ititle)}')

另外，请注意我使用的是 f-string，它需要 python 3.6+