python - 在 Python 中绘制不规则间隔的 RGB 图像

Question

我想用 matplotlib 在一组轴上绘制多个 RGB 图像。我最初尝试为此使用 imshow，但它似乎无法处理一组轴上具有不同范围的两个图像(当我绘制第二个时，它会使第一个消失)。我认为这里的答案是使用 pcolormesh，如 How to plot an irregular spaced RGB image using python and basemap?

但是，这似乎对我不起作用 - 来自可映射对象的颜色(即我传递给 pcolormesh 的数据数组，以及我指定的 cmap)会覆盖我指定的面颜色。网格的边缘确实有正确的颜色，但面不是。

有谁知道如何直接设置脸的颜色？理想情况下，我会使用采用 n*m*3 数组的 pcolormesh 作为 n*m 的替代方案，就像 imshow...

我希望它如何工作的一个最小示例: 将 numpy 导入为 np 将 matplotlib.pyplot 导入为 plt

#make some sample data
x, y = np.meshgrid(np.linspace(0,255,1),np.linspace(0,255,1))
r, g = np.meshgrid(np.linspace(0,255,100),np.linspace(0,255,120))
b=255-r

#this is now an RGB array, 100x100x3 that I want to display
rgb = np.array([r,g,b]).T

m = plt.pcolormesh(x, y, rgb/255.0, linewidth=0)
plt.show()

问题是对 plt.pcolormesh 的调用失败，

numRows, numCols = C.shape
ValueError: too many values to unpack

我猜这是因为它需要一个 2D 数组，而不是最后一个维度为 3 长的 3D 数组。

Answer 1

解决方案非常简单:我需要做的一件事不在我链接到的问题中，就是从网格对象中删除数组。一个最小的例子，以防对其他人有帮助:

import numpy as np
import matplotlib.pyplot as plt

#make some sample data
r, g = np.meshgrid(np.linspace(0,255,100),np.linspace(0,255,100))
b=255-r

#this is now an RGB array, 100x100x3 that I want to display
rgb = np.array([r,g,b]).T

color_tuple = rgb.transpose((1,0,2)).reshape((rgb.shape[0]*rgb.shape[1],rgb.shape[2]))/255.0

m = plt.pcolormesh(r, color=color_tuple, linewidth=0)
m.set_array(None)
plt.show()

我猜 color_tuple 行可能不明显:本质上我们需要将 (n, m, 3) 数组转换为 (n*m, 3) 数组。我认为转置对于让所有内容正确匹配是必要的。还值得注意的是，我们传入的颜色需要是 float ，介于 0 和 1 之间。