python - 绘制带有 alpha 值的数据，生成一张彩色图像 - Matplotlib

Question

我正在尝试使用不透明度值绘制某些数据。

要在不透明的情况下绘制它，我可以这样做：

plt.figure()
fig = plt.imshow(scalar_field, origin='lower', zorder=1, extent=(-4, 4, -4, 4))
plt.colorbar(fig)
plt.show()
plt.close()

这给了我那个图像：

但我有一些不透明度值应该包括在内。然后，我执行以下操作：

plt.figure()
data = plt.cm.jet(scalar_field)
data[..., 3] = 1.0 # just for tests purposes
fig = plt.imshow(data, origin='lower', extent=(-4, 4, -4, 4))
plt.colorbar(fig)
plt.show()
plt.close()

这是正确的，因为vmin=0.0andvmax=1.0和我的值太小了。

所以，为了正确地看到一些东西，我这样做：

plt.figure()
data = plt.cm.jet(scalar_field)
data[..., 3] = 1.0 # just for tests purposes
max_lim = numpy.max(field_property)
min_lim = numpy.min(field_property)
fig = plt.imshow(data, origin='lower', extent=(-4, 4, -4, 4), vmin=min_lim, vmax=max_lim, cmap=plt.cm.jet))
plt.colorbar(fig)
plt.show()
plt.close()

正确调整图像限制，如第一张图所示：

但是，我还看不到图像。在这两种情况下，我都可以在图像中看到一个奇怪的蓝色圆圈。

我错过了什么？

谢谢你。

Answer 1

您缺少的是cmap使用某些值调用 a 假定固定值的事实。如果传递 a float，则 cmap 的范围固定在 0 和 1 之间。如果传递 a int，范围是 0 到 255。显然，您将非常小的值传递给cmap然后返回等于或接近值 0 的颜色（所以蓝色的喷气机）。

然后，您将此“颜色数组”传递给 imshow，这假定并将数组显示为 RGB(A) 图像。您添加的任何vmin,vmax和cmap东西都会被忽略，因为它已经是一个 RGB 图像，没有什么可以映射的。

您要做的是创建一个ScalarMappable您可以使用设置范围的cmapAND创建的。normalizer

由于您没有定义“scalar_field”，我将使用 MPL Gallery 中的一些示例数据：

import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np

# create the sample data
# after http://matplotlib.org/examples/images_contours_and_fields/pcolormesh_levels.html
dx, dy = 0.05, 0.05
y, x = np.mgrid[slice(1, 5 + dy, dy),
                slice(1, 5 + dx, dx)]

z = np.sin(x) ** 10 + np.cos(10 + y * x) * np.cos(x)


fig, ax = plt.subplots(figsize=(6,6))
ax.axis('off')

# with a normalizer you can set the range against which you want to map
norm = mpl.colors.Normalize(vmin=-1, vmax=1)
# the cmap of your choice
cmap = plt.cm.jet

# create a 'mapper'
mapper = mpl.cm.ScalarMappable(norm=norm, cmap=cmap)

# map the data to a rgba array
rgba = mapper.to_rgba(z)

# make some adjustments to the alpha
rgba[:,:,3] = 0.5

# and plot
# there is no need for vmin, cmap etc keywords, its already rgba
ax.imshow(rgba)