for 循环plt.subplot画图
时间: 2023-08-19 13:11:38 浏览: 136
巧用for循环绘制各种图形
你可以使用for循环结合plt.subplot来画图。有几种方法可以实现这个目标。
第一种方法是创建子图对象然后进行绘制。你可以使用plt.subplot来创建子图对象,然后在每个子图对象上进行绘制。例如,你可以使用以下代码:
```python
import matplotlib.pyplot as plt
fig = plt.figure()
for i in range(1, 5):
ax = plt.subplot(2, 2, i)
ax.plot([1, 2, 3])
plt.show()
```
这个例子中,我们创建了一个包含2行2列的子图对象,然后使用for循环在每个子图对象上进行绘制。
第二种方法是同时创建figure和子图对象。你可以使用plt.subplots来同时创建figure和子图对象,并使用一个嵌套的for循环在每个子图对象上进行绘制。例如,你可以使用以下代码:
```python
import matplotlib.pyplot as plt
fig, axes = plt.subplots(2, 2)
for i in range(2):
for j in range(2):
axes[i, j].plot([1, 2, 3])
plt.show()
```
这个例子中,我们使用plt.subplots创建了一个包含2行2列的子图对象,并使用嵌套的for循环在每个子图对象上进行绘制。
第三种方法是使用一个列表来存储子图对象,然后使用for循环在每个子图对象上进行绘制。例如,你可以使用以下代码:
```python
import matplotlib.pyplot as plt
fig, axes = plt.subplots(3, 3)
axes_list = []
for i in range(axes.shape[0]):
for j in range(axes.shape[1]):
axes_list.append(axes[i, j])
for ax in axes_list:
ax.plot([1, 2, 3])
plt.show()
```
这个例子中,我们使用plt.subplots创建了一个包含3行3列的子图对象,并使用一个列表来存储子图对象。然后,我们使用for循环在每个子图对象上进行绘制。
无论你选择哪种方法,都可以使用for循环结合plt.subplot来画图。希望这些例子对你有帮助!
阅读全文