我可以用文本覆盖做类似 imsave() 的事情吗？

Question

我按顺序使用imsave() 制作了许多PNG，我将它们组合成一个AVI，我想添加移动文本注释。我使用ImageJ 制作 AVI 或 GIF。

我不想要坐标轴、数字、边框或任何东西，只想要带有文本（可能还有箭头）的彩色图像（例如imsave() 提供的）。这些将逐帧更改。请原谅使用喷气机。

我可以使用 savefig() 并关闭标记，然后进行裁剪作为后期处理，但是有没有更方便、直接或“matplotlibithic”的方式来做到这一点，这对我的硬盘来说不会那么难？ （最后的事情会非常大）。

一个代码sn-p，由请求添加：

import numpy as np
import matplotlib.pyplot as plt

nx, ny = 101, 101

phi   = np.zeros((ny, nx), dtype = 'float')
do_me = np.ones_like(phi, dtype='bool')

x0, y0, r0 = 40, 65, 12

x = np.arange(nx, dtype = 'float')[None,:]
y = np.arange(ny, dtype = 'float')[:,None]
rsq = (x-x0)**2 + (y-y0)**2

circle = rsq <= r0**2

phi[circle] = 1.0
do_me[circle] = False

do_me[0,:], do_me[-1,:], do_me[:,0], do_me[:,-1] = False, False, False, False

n, nper = 100, 100
phi_hold = np.zeros((n+1, ny, nx))
phi_hold[0] = phi

for i in range(n):

    for j in range(nper):
        phi2 = 0.25*(np.roll(phi,  1, axis=0) +
                     np.roll(phi, -1, axis=0) +
                     np.roll(phi,  1, axis=1) +
                     np.roll(phi, -1, axis=1) )

        phi[do_me] = phi2[do_me]

    phi_hold[i+1] = phi

change = phi_hold[1:] - phi_hold[:-1]

places = [(32, 20), (54,25), (11,32), (3, 12)]

plt.figure()
plt.imshow(change[50])
for (x, y) in places:
    plt.text(x, y, "WOW", fontsize=16)
plt.text(5, 95, "Don't use Jet!", color="white", fontsize=20)
plt.show()

Answer 1

方法一

使用an excellent answer to another question 作为参考，我想出了以下简化的变体，它似乎工作得很好——只要确保figsize (which is given in inches) 的纵横比与绘图数据的大小比相匹配：

import numpy as np
import matplotlib.pyplot as plt

test_image = np.eye(100)
fig = plt.figure(figsize=(4,4))
ax = plt.axes(frameon=False, xticks=[],yticks=[])
ax.imshow(test_image)
plt.savefig('test.png', bbox_inches='tight', pad_inches=0)

请注意，我将 imshow 与 test_image 一起使用，这可能与其他绘图功能不同...如果您想做其他事情，请在评论中告诉我。

还要注意，图像将被（重新）采样，因此figsize 会影响写入图像的分辨率。

作为pointed out in the comments，figsize 设置与输出图像的大小（或屏幕上的大小，就此而言）不匹配。为了克服这个问题，使用...

方法二

阅读常见问题解答条目Move the edge of an axes to make room for tick labels，我找到了一种方法，可以让figsize 参数直接设置输出图像大小，方法是将轴的刻度移出可见区域：

import numpy as np
import matplotlib.pyplot as plt

test_image = np.eye(100)
fig = plt.figure(figsize=(4,4))
ax = fig.add_axes([0,0,1,1])
ax.imshow(test_image)
plt.savefig('test.png')

请注意，savefig 具有默认 DPI 设置（在我的情况下为 100），它与 figsize 结合使用 - 确定保存图像的 x 和 y 方向的像素数。你可以用dpi 关键字参数覆盖它savefig。

如果你想在屏幕上显示图像而不是保存它（通过使用plt.show()而不是上面代码中的plt.savefig行），图形的大小取决于（除了已经熟悉的@ 987654341@ 参数）figure 的 DPI 设置，它也有默认值（我的系统上为 80）。可以通过将 dpi 关键字参数传递给 plt.figure() 调用来覆盖此值。