Python| 水文 |批量读取NC文件降水数据并导出为Excel相应格式

import xarray as xr
import pandas as pd
import numpy as np
import os
 
def read_precip_files(file_paths, var_name='precip'):
    """
    批量读取降水数据NC文件并合并为一个DataFrame。
    :param file_paths: 文件路径列表
    :param var_name: 降水数据变量名，默认为'precip'
    :return: 合并后的DataFrame
    """
    df_list = []
    for file_path in file_paths:
        ds = xr.open_dataset(file_path)
        df = ds[var_name].to_dataframe().reset_index()
        df['filename'] = os.path.basename(file_path)
        df_list.append(df)
    return pd.concat(df_list, ignore_index=True)
 
def export_to_excel(df, output_path, sheet_name='data'):
    """
    将DataFrame导出为Excel文件。
    :param df: 待导出的DataFrame
    :param output_path: Excel文件的路径
    :param sheet_name: Excel中的工作表名称
    """
    with pd.ExcelWriter(output_path) as writer:
        df.to_excel(writer, sheet_name=sheet_name, index=False)
 
# 示例使用
file_paths = ['path/to/precipitation_data_1.nc', 'path/to/precipitation_data_2.nc']
df = read_precip_files(file_paths)
export_to_excel(df, 'combined_precipitation_data.xlsx')

这段代码定义了两个函数：read_precip_files用于批量读取降水数据NC文件并合并为一个DataFrame；export_to_excel用于将DataFrame导出为Excel文件。示例使用部分展示了如何使用这两个函数。