ggplot如何为栅格图中不存在的整数Z值设置填充色并显示在图例

嘿，这个需求我熟！要给栅格图里没出现的Z值（比如你说的6）指定颜色并放进图例，咱们可以用Python的matplotlib库来实现，具体步骤我给你拆解清楚：

1. 先把数据整理成栅格格式 #

首先得把你给出的散点式X/Y/Z数据转换成二维网格（栅格）格式，这样才能绘制栅格图：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap
from matplotlib.patches import Patch

# 把你的原始数据转成数组
X = np.array([84,85,86,87,88,89,100,101,102,103,104,105,116])
Y = np.array([75.75,76.00,76.25,76.50,76.75,77.00,75.75,76.00,76.25,76.50,76.75,77.00,75.75])
Z = np.array([7,5,5,2,7,4,2,4,2,3,2,4,1])

# 提取唯一的经纬度值，构建网格框架
unique_lon = np.unique(X)
unique_lat = np.unique(Y)
grid_z = np.zeros((len(unique_lat), len(unique_lon)))

# 把Z值填充到对应的网格位置
for lon, lat, z_val in zip(X, Y, Z):
    lon_idx = np.where(unique_lon == lon)[0][0]
    lat_idx = np.where(unique_lat == lat)[0][0]
    grid_z[lat_idx, lon_idx] = z_val

2. 自定义包含Z=6的颜色映射 #

接下来我们要给所有需要展示的Z值（包括没出现的6）都指定颜色，用ListedColormap来创建精准的颜色映射：

# 定义所有要展示的Z值范围（包含未出现的6）
all_z_vals = [1,2,3,4,5,6,7]
# 给每个Z值分配颜色，你可以根据自己的需求修改
color_list = [
    '#f8d5e4',  # Z=1
    '#c6e2ff',  # Z=2
    '#d0f0c0',  # Z=3
    '#fff2cc',  # Z=4
    '#e6cbeb',  # Z=5
    '#ffb380',  # Z=6（专门给未出现的值指定的颜色）
    '#ff6b6b'   # Z=7
]

# 创建自定义颜色映射
custom_cmap = ListedColormap(color_list)
# 设置颜色归一化范围，确保覆盖所有要展示的Z值
norm = plt.Normalize(min(all_z_vals), max(all_z_vals))

3. 绘制栅格图并添加包含Z=6的图例 #

默认的图例只会显示数据中存在的Z值，所以我们要手动创建图例条目，把Z=6加进去：

plt.figure(figsize=(10,7))
# 绘制栅格图
im = plt.pcolormesh(unique_lon, unique_lat, grid_z, cmap=custom_cmap, norm=norm, edgecolors='white')

# 手动创建每个Z值的图例块
legend_patches = [Patch(facecolor=color_list[i], label=str(val)) for i, val in enumerate(all_z_vals)]

# 添加图例，位置可以自己调整
plt.legend(handles=legend_patches, title="Z 值", bbox_to_anchor=(1.02, 1), loc='upper left')

# 设置图表标签和标题
plt.xlabel('经度 (X)')
plt.ylabel('纬度 (Y)')
plt.title('包含未出现Z=6的栅格图')
# 可选：添加颜色条，颜色条也会显示Z=6的颜色
plt.colorbar(im, ticks=all_z_vals)

plt.tight_layout()
plt.show()

关键细节说明 #

• 如果你的数据是不规则散点，不是规则栅格，那需要先用scipy.interpolate做插值处理，把数据转成规则网格再绘制。
• 颜色和图例位置都可以根据你的需求自由调整，只要保证color_list和all_z_vals的顺序对应上就行。

内容的提问来源于stack exchange，提问作者user1142937