Python - 基于值绘制网格

Question

我检查了类似的线程，我的问题将更进一步:Plotting colored grid based on values

我有一个 20 x 10 的网格，其中第一个单元格(左下角)的 ID = 0，最后一个单元格(右上角)的 ID = 99。假设我有两个列表。第一个是值大于 0 的单元格列表，第二个列表由这些值组成，例如ID = 11 的单元格的值为 77。

Cellid = [2, 4 ,5, 11 ,45 ,48 ,98]
Cellval = [20, 45 ,55, 77,45 ,30 ,15]

1. 我想绘制此网格的二维图，其中单元格颜色为基于值(value)。空单元格为空白，其余部分:值越大，单元格越绿。
2. 其次，我想将 2D 图从之前的点升级为 3D 图，其中单元格值是列。

你能给我一个如何处理这个问题的建议吗？

Answer 1

“grid size 20 x 10”和右上ID=99有矛盾。所以我假设您在这里指的是“网格尺寸 10 x 10”。

然后您可以创建一个数组，除了 Cellid 给出的位置外，所有位置都为 0。在这里，我假设 ID 首先沿着 x 运行。您可以屏蔽它，使 0 完全不被着色；他们将其绘制为 imshow。

import numpy as np
import matplotlib.pyplot as plt

nrows = 10
ncols = 10

Cellid = [2, 4 ,5, 11 ,45 ,48 ,98]
Cellval = [20, 45 ,55, 77,45 ,30 ,15]

data = np.zeros(nrows*ncols)
data[Cellid] = Cellval
data = np.ma.array(data.reshape((nrows, ncols)), mask=data==0)

fig, ax = plt.subplots()
ax.imshow(data, cmap="Greens", origin="lower", vmin=0)

# optionally add grid
ax.set_xticks(np.arange(ncols+1)-0.5, minor=True)
ax.set_yticks(np.arange(nrows+1)-0.5, minor=True)
ax.grid(which="minor")
ax.tick_params(which="minor", size=0)

plt.show()