Python NetCDF4读取与保存nc文件

Python NetCDF4读取与保存nc文件

# -*- coding: utf-8 -*-
import netCDF4 as nc
'''
old_file_name:用来读取
new_file_name:用来写入

读取步骤：
    1、概览nc文件内容，先知晓有哪些变量
    2、按变量名取出值
    
写入步骤：
    1、创建维度      （相当于创建坐标轴）
    2、创建变量
    3、填充维度的值    （相当于给坐标轴加上刻度）
    4、填充变量值
'''
def read_write_nc(new_file_name="testWrite.nc", old_file_name="testRead.nc"):
    # 用于读取数据
    old_NC = nc.Dataset(old_file_name)
    # 获取维度的值，一般有时间、等级、经纬度
    time = old_NC.variables['time'][:].data
    level = old_NC.variables['level'][:].data
    latitude = old_NC.variables['latitude'][:].data
    longitude = old_NC.variables['longitude'][:].data

    # 获取变量值
    z = old_NC.variables['z'][:].data

    # 以上为读取数据，深度学习的某些数据存在nc文件中比较方便

    # 以下为写入数据
    new_NC = nc.Dataset(new_file_name, 'w', format='NETCDF4')
    # 创建维度，createDimension（维度名称，维度长度）
    new_NC.createDimension('time', len(time))
    new_NC.createDimension('level', len(level))
    new_NC.createDimension('latitude', len(latitude))
    new_NC.createDimension('longitude', len(longitude))

    # 创建变量，createVariable（变量名，值类型，shape）
    # shape里面用上面创建的维度来约束，维度也是一种变量，一般是一维的
    new_NC.createVariable("time", 'u4', ("time"))# ("time")表示维度由刚刚创建的'time'维度来约束
    new_NC.createVariable("level", 'f', ("level"))
    new_NC.createVariable("latitude", 'f', ("latitude"))
    new_NC.createVariable("longitude", 'f', ("longitude"))

    # 创建变量，z、r、t等变量为四维数组，维度分别为时间、等级、经纬度
    new_NC.createVariable("z", 'f', ("time", "level", "latitude", "longitude"))

    # 给维度填充数据
    # 偷懒，用刚刚读出的数据来填充
    # 相当于给数轴表上刻度
    new_NC.variables['time'][:] = time
    new_NC.variables['level'][:] = level
    new_NC.variables['latitude'][:] = latitude
    new_NC.variables['longitude'][:] = longitude

    # 给变量填充数据
    new_NC.variables['z'][:] = z

    # 关闭文件
    new_NC.close()

快速概览nc文件