plt.subplots(),imshow画图时,我想添加额外的轴
时间: 2023-11-02 11:09:10 浏览: 33
要添加额外的轴,可以使用`plt.axes()`函数。该函数会创建一个新的坐标轴对象,然后您可以使用该对象在图形中添加自定义元素,如文本、箭头等。
以下是一个示例代码片段,展示如何在使用`imshow()`函数绘制图像时添加额外的轴:
```python
import matplotlib.pyplot as plt
import numpy as np
# 创建一个 2x2 的子图,并获取第一个子图的坐标轴对象
fig, axes = plt.subplots(2, 2)
ax = axes[0, 0]
# 加载一张图像
img = np.random.rand(10, 10)
# 在图像上绘制热力图,并将图像的坐标轴对象传递给 imshow() 函数
im = ax.imshow(img)
# 创建一个新的坐标轴对象,并添加到图形上
ax2 = plt.axes([0.6, 0.6, 0.2, 0.2])
ax2.plot([1, 2, 3], [4, 5, 6])
# 显示图形
plt.show()
```
在这个例子中,我们首先使用`plt.subplots()`函数创建了一个 2x2 的子图,然后获取了第一个子图的坐标轴对象。接下来,我们加载了一张随机的图像,并使用`imshow()`函数在坐标轴对象上绘制了热力图。然后,我们使用`plt.axes()`函数创建了一个新的坐标轴对象,并指定了它的位置和大小。最后,我们在新的坐标轴对象上绘制了一条线,并使用`plt.show()`函数显示了整个图形。
您可以根据自己的需要修改新坐标轴的位置和大小,并在上面添加任何自定义元素。