如何在Python中为DataFrame设置经纬度行列索引并转为xarray数据集

1. 导入依赖库 #

import numpy as np
import pandas as pd
import xarray as xr

2. 生成目标经纬度数组 #

按需求生成指定范围和精度的经纬度序列：

longitude = np.arange(-180, 180, 0.05)
latitude = np.arange(-90, 90, 0.05)

3. 重构DataFrame的索引与列 #

假设你的原始数据存储在名为raw_df的DataFrame中，通过reindex方法将行索引替换为纬度数组、列索引替换为经度数组——该操作会自动对齐原始数据中的有效值，缺失位置填充NaN，完美匹配你原数据的情况：

# 重新索引，对齐经纬度并补全缺失值
processed_df = raw_df.reindex(index=latitude, columns=longitude)

4. 转换为xarray数据集 #

使用from_dataframe方法将处理后的DataFrame转为xarray数据集，同时指定维度名称为lat（纬度）和lon（经度）：

# 重命名轴标签后转换为xarray Dataset
xr_dataset = xr.Dataset.from_dataframe(processed_df.rename_axis(index='lat', columns='lon'))

确认结果 #

可通过打印数据集查看结构，验证经纬度坐标和数值变量是否正确：

print(xr_dataset)

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