plt.subplots()，imshow画图时，我想添加额外的轴在imshow里，并加上刻度

可以使用`plt.gca()`获取当前的坐标轴对象，然后调用`ax.spines`方法添加额外的轴线，再用`ax.tick_params`方法设置刻度。以下是一个例子：

import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots()

# 生成数据
x = np.arange(-5, 5, 0.1)
y = np.sin(x)

# 绘制图像
im = ax.imshow([[1]], cmap='gray', extent=[x.min(), x.max(), y.min(), y.max()])
ax.plot(x, y)

# 添加额外的轴线
ax.spines['top'].set_visible(True)
ax.spines['right'].set_visible(True)

# 设置刻度
ax.tick_params(axis='both', which='both', direction='in')

plt.show()

其中，`ax.spines`方法可以传入参数来选择要添加的轴线，例如`ax.spines['top']`表示添加顶部轴线，`ax.spines['right']`表示添加右侧轴线。`ax.tick_params`方法可以用来设置刻度的方向和样式。

相关问题

plt.subplots()，imshow画图时，我想添加额外的轴在imshow里

可以使用 matplotlib 的 `Axes` 对象来实现在 `imshow` 里添加额外的轴。首先，你需要创建一个 `Axes` 对象，并将其添加到 `imshow` 的返回值中。然后，在 `Axes` 对象上，你可以使用 matplotlib 提供的各种方法来添加额外的轴（例如，使用 `axhline` 和 `axvline` 方法添加水平和垂直线）。

以下是一个简单的示例，展示如何在 `imshow` 中添加额外的轴：

import numpy as np
import matplotlib.pyplot as plt

# 生成一张图片
img = np.random.rand(10, 10)

# 创建一个包含两个 Axes 对象的 figure
fig, (ax1, ax2) = plt.subplots(ncols=2)

# 在第一个 Axes 对象中显示图片
im = ax1.imshow(img)

# 在第二个 Axes 对象中添加一个水平线和一个垂直线
ax2.axhline(y=5, color='r')
ax2.axvline(x=5, color='r')

# 将第一个 Axes 对象的 colorbar 移动到第二个 Axes 对象的位置
cbar = fig.colorbar(im, ax=ax2)

plt.show()

运行代码后，你应该会看到一张包含两个子图的图片：左边的子图是一个 `imshow` 显示的图片，右边的子图添加了一条水平线和一条垂直线。此外，右边的子图还包含一个 colorbar，它实际上属于左边的子图，但被移动到了右边的子图的位置上。

plt.subplots()，imshow画图时，我想添加额外的轴

要添加额外的轴，可以使用`plt.axes()`函数。该函数会创建一个新的坐标轴对象，然后您可以使用该对象在图形中添加自定义元素，如文本、箭头等。

以下是一个示例代码片段，展示如何在使用`imshow()`函数绘制图像时添加额外的轴：

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()`函数显示了整个图形。

您可以根据自己的需要修改新坐标轴的位置和大小，并在上面添加任何自定义元素。