如何在matplotlib中绘制带有标准化颜色的点？

Question

我有以下代码：

x = range(1, 100) # pixel x coordinates
y = range(21, 120) # pixel y coordinates
z = [-2, 5, 1, ..., 1] # should be mapped to colors

import matplotlib.pyplot as plt
import matplotlib.cm as cm
from matplotlib.colors import Normalize

cmap = cm.autumn
norm = Normalize(vmin=min(z), vmax=max(z))
colormap = cmap(norm(5))
plt.plot(x, y, cm=colormap)

现在我想绘制这些点[x, y]这样：

1. 每个点应该是一个像素（因为len(x)可能是大约35'000 ），而不仅仅是一个标记（我担心它们会在其他方面重叠）。
2. 使用这个问题 ，我想拍摄一张色图并将min(z)映射为白色和max(z)为黑色（并显示图例）。

我怎么能用matplotlib做到这一点？

Answer 1

既然你在谈论的是像素而不是标记，我不知道你是否得到了所有像素坐标的数据，或者仅仅是针对某些像素坐标，我会告诉你两者的内容。 我将包含numpy包，因为它使得使用数组的所有内容变得更容易。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm
from matplotlib.colors import Normalize

x = range(1, 100) # pixel x coordinates
y = range(21, 120) # pixel y coordinates

假设您有[x, y]所有坐标的z数据：

In [1]: z = np.random.rand(len(x), len(y)) # should be mapped to colors
plt.imshow(z)
plt.colorbar()
Out [1]:

假设每个z值都映射到[x, y]坐标对，但不是所有坐标都被覆盖：

In [2]:
z = np.random.rand(len(x))

arr = np.zeros([max(x) + 1, max(y) + 1])
arr[x, y] = z

plt.imshow(arr)
plt.colorbar()

Out [2]: