在 matplotlib 中使用 2 轴颜色图

Question

有没有一种实用的方法来使用/定义一个“漂亮”的 Matplotlib 颜色图，它取决于两个实变量 cmap2d(a1,a2)，当然例如：0.<= a1 <= 1. & 0.<= a2 <= 1.？

基本上它应该被定义为一个常规的颜色图，但定义在一个二维帧上......

Answer 1

是的，您可以定义 2D 颜色图。

因为您的 2D 颜色图的输出是单变量的（让我们暂时忘记 RGB 通道），您总是可以分两步来表达它：

1. 计算一个函数f(a1, a2)
2. 将函数的输出传递给您选择的颜色图。

鉴于上面的分解，您不必太担心颜色图是否需要一个或两个（或更多）参数，因为您总是可以用普通的一维颜色图来表达它。

如果您愿意将不同的渠道视为不同的变量，您可以执行以下操作：

import numpy as np
from matplotlib import pyplot as plt
# Create a grid of values
x = np.linspace(0, 1)
y = np.linspace(0, 1)
xx, yy = np.meshgrid(x, y)
# Apply some function
xx = xx ** 2
yy = np.sin(np.pi * yy)
# Plot as RGB values
img = np.transpose((xx, yy, np.zeros_like(xx)))
plt.imshow(img)