循环绘制子图

Question

z = {'A': [0.3618426, 0.36146951], 'B': [1.8908799, 1.904695], 'C': [2.1813462e+08, 2.1833622e+08], 'D': [0.89925492, 0.89953589], 'E': [2.6356747, 2.6317911], 'F': [2.2250445e+08, 2.2501808e+08], 'G': [2.0806053e+08, 2.0691238e+08], 'H': [0.37242803, 0.37611806]}
k = [1,2]

for key in z:
plt.subplot(4,4,1)
plt.plot(k,[z[key][0],z[key][1]], 'ro-')
plt.show()

我会努力弄清楚。 z是大小不同的字典。我想做的是绘制字典数量为4列的图，但行数应根据生成的图数来增加，例如，如果要绘制16个键，则应该以4行4列图形结尾。我可以这样做吗？

Answer 1

绘制多个图形的基本形式是以下方法。作为前提条件，您只需要确定列数即可。 cols=3 循环过程的其余部分由字典的数量完成。

import matplotlib.pyplot as plt
import pandas as pd
import math

z = {'A': [0.3618426, 0.36146951], 'B': [1.8908799, 1.904695], 'C': [2.1813462e+08, 2.1833622e+08], 'D': [0.89925492, 0.89953589], 'E': [2.6356747, 2.6317911], 'F': [2.2250445e+08, 2.2501808e+08], 'G': [2.0806053e+08, 2.0691238e+08], 'H': [0.37242803, 0.37611806]}
k = [1,2]

cols = 3
rows = math.ceil(len(z) / cols)

fig, axes = plt.subplots(rows, cols, figsize=(16,12))

dict_keys = [k for k in z.keys()]

l = 0
for i in range(rows):
    for j in range(cols):
        if len(z) == l:
            break
        else:
            key = dict_keys[i+j]
            axes[i][j].plot(k, [z[key][0],z[key][1]], 'ro-')
        l += 1

plt.show()

Answer 2

以下代码可以为您提供帮助：

from math import ceil

# 9 elements in my dict (4x4 + 1)
z = {'A': [0.3618426, 0.36146951], 'B': [1.8908799, 1.904695], 'C': [2.1813462e+08, 2.1833622e+08], 'D': [0.89925492, 0.89953589], 'E': [2.6356747, 2.6317911], 'F': [2.2250445e+08, 2.2501808e+08], 'G': [2.0806053e+08, 2.0691238e+08], 'H': [0.37242803, 0.37611806], 'X': [0.37242803, 0.37611806]}
k = [1,2]

plt.subplots(figsize=(16,8)) # optional

# fixed number of columns
cols = 4
# number of rows, based on cols
rows = ceil(len(z) / cols)

# iterate through indices and keys
for index, key in enumerate(z):
    # new subplot with (i + 1)-th index laying on a grid
    plt.subplot(rows, cols, index + 1) 
    # drawing the plot
    plt.plot(k, [z[key][0], z[key][1]], 'ro-')

# render everything
plt.show()