在 Python 中读取 netCDF 文件时用户输入文件名

Question

我有一组从 1953 年到 2014 年的土壤水分数据文件。它们都是 cpc_soil_YYYY.nc 的形式（其中 YYYY 是那些年份之一）。有没有办法让我要求用户输入用户想要查看的年份，并让我的程序打开相应的功能？我目前有它，我必须在 gedit 中手动更改年份，并编写函数来获取每个变量（土壤湿度作为时间、纬度、经度的函数）：

    import netCDF4 as nc
    import numpy as np
    import numpy.ma as ma
    import csv as csv

    fid=nc.MFDataset('/data/reu_data/soil_moisture/cpc_soil_1957.nc','r')
    fid.close()
    ncf='/data/reu_data/soil_moisture/cpc_soil_1957.nc'

    def read_var(ncfile, varname):
        fid=nc.Dataset(ncfile, 'r')
        out=fid.variables[varname][:]   
        fid.close()
        return out

    time=read_var(ncf, 'time')
    lat=read_var(ncf, 'lat')
    lon=read_var(ncf, 'lon')
    soil=read_var(ncf, 'soilw')

Answer 1

您可以使用 input() 要求用户输入年份。然后你可以用它来生成文件路径。

...
year = input("Enter year: "))
filename = '/data/reu_data/soil_moisture/cpc_soil_%s.nc' % (year,)
fid=nc.MFDataset(filename,'r')
fid.close()
...

您应该进行错误检查以确保用户输入的值实际上是一年并且在您的数据范围内。

您可以在 Python here 中阅读有关 input/output 的更多信息。

在 Python 中读取 netCDF 文件时用户输入文件名

User input filename when reading in netCDF files in Python

python

raw-input

netcdf