如何将Python绘图设置为4×2布局？现有8行1列绘图代码优化求助

嘿，要把8行1列的布局改成4×2其实很容易，只需要修改几个关键地方就行，我给你拆解一下：

核心修改点 #

• 修改plt.subplots的布局参数
  原来的代码里plt.subplots(8,1)是创建8行1列的子图，你只需要把参数改成plt.subplots(4, 2)，这样就能生成4行2列的布局啦。

• 调整子图变量的解构方式
  原来的变量是按8个单独的子图依次排列，现在4×2的布局可以用嵌套的元组来解构，或者直接用一个二维数组来管理子图，后者更方便批量操作。比如：

  • 如果你想继续单独命名每个子图，可以写成：

    fig, ((ax1, ax2), (ax3, ax4), (ax5, ax6), (ax7, ax8)) = plt.subplots(4, 2)
    

    这样ax1是第一行第一列，ax2是第一行第二列，以此类推。
  • 更高效的方式是直接用一个二维数组接收子图：

    fig, axes = plt.subplots(4, 2)
    

    之后可以通过axes[0,0]访问第一行第一列的子图，axes[0,1]访问第一行第二列的子图，以此类推，还能通过axes.flat遍历所有子图，省去重复写8次plot的麻烦。

• 批量设置坐标轴范围
  原来的plt.axis([0, 1, -1, 1])只会对最后一个子图生效，所以最好批量给所有子图设置轴范围。比如用plt.setp来一次性设置：

  plt.setp(axes, xlim=(0, 1), ylim=(-1, 1))
  

  或者遍历每个子图单独设置。

修改后的完整代码示例 #

这里给你两种写法，你可以根据需求选：

写法1：保留单独子图命名

import numpy as np
import pylab as plt

t = np.arange(0.0, 1.0, 0.01)
s = np.sin(2*2*np.pi*t)
# 修改布局参数为4行2列，解构对应位置的子图
fig, ((ax1, ax2), (ax3, ax4), (ax5, ax6), (ax7, ax8)) = plt.subplots(4, 2)

# 给每个子图绘图
ax1.plot(t, s, 'o', color="red")
ax2.plot(t, s, 'o', color="red")
ax3.plot(t, s, 'o', color="red")
ax4.plot(t, s, 'o', color="red")
ax5.plot(t, s, 'o', color="red")
ax6.plot(t, s, 'o', color="red")
ax7.plot(t, s, 'o', color="red")
ax8.plot(t, s, 'o', color="red")

# 批量设置所有子图的坐标轴范围
plt.setp(fig.axes, xlim=(0, 1), ylim=(-1, 1))
plt.tight_layout() # 自动调整子图间距，避免重叠
plt.show()

写法2：用数组批量操作（更简洁）

import numpy as np
import pylab as plt

t = np.arange(0.0, 1.0, 0.01)
s = np.sin(2*2*np.pi*t)
# 创建4行2列的子图，得到二维数组axes
fig, axes = plt.subplots(4, 2)

# 遍历所有子图，批量绘图
for ax in axes.flat:
    ax.plot(t, s, 'o', color="red")

# 批量设置坐标轴范围
plt.setp(axes, xlim=(0, 1), ylim=(-1, 1))
plt.tight_layout()
plt.show()

这样修改之后，你的子图就会以4×2的布局整齐排列啦，plt.tight_layout()还能自动调整子图之间的间距，避免标签重叠哦～

内容的提问来源于stack exchange，提问作者Peter Harris