将csv转换为netcdf

时间:2014-04-08 10:10:06

标签: python csv netcdf

我正在尝试通过Python将.csv文件转换为netCDF4,但我无法弄清楚如何将.csv表格格式的信息存储到netCDF中。我主要担心的是我们如何将列中的变量声明为可行的netCDF4格式?我发现的一切通常都是从netCDF4中提取信息到.csv或ASCII。我提供了样本数据,示例代码和我的错误,用于声明适当的数组。任何帮助将不胜感激。

示例表如下:

Station Name    Country  Code   Lat Lon mn.yr   temp1   temp2   temp3   hpa 
Somewhere   US  12340   35.52   23.358  1.19    -8.3    -13.1   -5  69.5
Somewhere   US  12340           2.1971  -10.7   -13.9   -7.9    27.9
Somewhere   US  12340           3.1971  -8.4    -13 -4.3    90.8

我的示例代码是:

#!/ usr / bin / env python

import scipy
import numpy
import netCDF4
import csv

from numpy import arange, dtype 

#Declare empty arrays

v1 = []
v2 = []
v3 = []
v4 = []

#打开csv文件并声明每个标题的数组变量

f = open('station_data.csv', 'r').readlines()

for line in f[1:]:
    fields = line.split(',')
    v1.append(fields[0]) #station
    v2.append(fields[1])#country
    v3.append(int(fields[2]))#code
    v4.append(float(fields[3]))#lat
    v5.append(float(fields[3]))#lon
#more variables included but this is just an abridged list
print v1
print v2
print v3
print v4

#convert to netcdf4 framework as netcdf

ncout = netCDF4.Dataset('station_data.nc','w') 

#纬度和经度。包含缺少号码的NaN

lats_out = -25.0 + 5.0*arange(v4,dtype='float32')
lons_out = -125.0 + 5.0*arange(v5,dtype='float32')

#输出数据。

press_out = 900. + arange(v4*v5,dtype='float32') # 1d array
press_out.shape = (v4,v5) # reshape to 2d array
temp_out = 9. + 0.25*arange(v4*v5,dtype='float32') # 1d array
temp_out.shape = (v4,v5) # reshape to 2d array

#创建lat和lon维度。

ncout.createDimension('latitude',v4)
ncout.createDimension('longitude',v5)

#定义坐标变量。他们将保留坐标信息

lats = ncout.createVariable('latitude',dtype('float32').char,('latitude',))
lons = ncout.createVariable('longitude',dtype('float32').char,('longitude',))

#分配单位属性以协调var数据。这会将text属性附加到包含单位的每个坐标变量。

lats.units = 'degrees_north'
lons.units = 'degrees_east'

#将数据写入协调变量。

lats[:] = lats_out
lons[:] = lons_out

#创建压力和温度变量

press = ncout.createVariable('pressure',dtype('float32').char,('latitude','longitude'))
temp = ncout.createVariable('temperature',dtype('float32').char,'latitude','longitude'))

#设置单位属性。

press.units =  'hPa'
temp.units = 'celsius'

#将数据写入变量。

press[:] = press_out
temp[:] = temp_out

ncout.close()
f.close()

错误:

Traceback (most recent call last):
  File "station_data.py", line 33, in <module>
    v4.append(float(fields[3]))#lat
ValueError: could not convert string to float: 

3 个答案:

答案 0 :(得分:10)

这是xarray的完美工作,这是一个python包,其数据集对象代表netcdf公共数据模型。以下是您可以尝试的示例:

import pandas as pd
import xarray as xr

url = 'http://www.cpc.ncep.noaa.gov/products/precip/CWlink/'

ao_file = url + 'daily_ao_index/monthly.ao.index.b50.current.ascii'
nao_file = url + 'pna/norm.nao.monthly.b5001.current.ascii'

kw = dict(sep='\s*', parse_dates={'dates': [0, 1]},
          header=None, index_col=0, squeeze=True, engine='python')

# read into Pandas Series
s1 = pd.read_csv(ao_file, **kw)
s2 = pd.read_csv(nao_file, **kw)

s1.name='AO'
s2.name='NAO'

# concatenate two Pandas Series into a Pandas DataFrame
df=pd.concat([s1, s2], axis=1)

# create xarray Dataset from Pandas DataFrame
xds = xr.Dataset.from_dataframe(df)

# add variable attribute metadata
xds['AO'].attrs={'units':'1', 'long_name':'Arctic Oscillation'}
xds['NAO'].attrs={'units':'1', 'long_name':'North Atlantic Oscillation'}

# add global attribute metadata
xds.attrs={'Conventions':'CF-1.0', 'title':'AO and NAO', 'summary':'Arctic and North Atlantic Oscillation Indices'}

# save to netCDF
xds.to_netcdf('/usgs/data2/notebook/data/ao_and_nao.nc')

然后运行ncdump -h ao_and_nao.nc会产生:

netcdf ao_and_nao {
dimensions:
        dates = 782 ;
variables:
        double dates(dates) ;
                dates:units = "days since 1950-01-06 00:00:00" ;
                dates:calendar = "proleptic_gregorian" ;
        double NAO(dates) ;
                NAO:units = "1" ;
                NAO:long_name = "North Atlantic Oscillation" ;
        double AO(dates) ;
                AO:units = "1" ;
                AO:long_name = "Arctic Oscillation" ;

// global attributes:
                :title = "AO and NAO" ;
                :summary = "Arctic and North Atlantic Oscillation Indices" ;
                :Conventions = "CF-1.0" ;

请注意,您可以使用xarray安装pip,但如果您使用的是Anaconda Python Distribution,则可以使用以下命令从Anaconda.org/conda-forge频道安装:

conda install -c conda-forge xarray

答案 1 :(得分:2)

xarray是一个很好的候选人,但我认为iris更好,因为它会在您犯错时通过引发错误来帮助您处理CF约定。

下面的笔记本重新实现了AO / NOA示例:

http://nbviewer.ipython.org/gist/ocefpaf/c66a7d0b967664ee4f5c

(请参阅底部的最后一个单元格,了解&#34; CF-convention优点&#34;虹膜。)

答案 2 :(得分:0)

如果您看到输入文件,则第二行中的列 Lat 没有对应的值。 当您读取csv文件时,此值fields[3]将存储为空字符串""。这就是你获得ValueError的原因。 您可以定义一个可以处理此错误的新函数,而不是使用默认函数:

def str_to_float(str):
    try:
        number = float(str)
    except ValueError:
        number = 0.0
# you can assign an appropriate value instead of 0.0 which suits your requirement
    return number

现在你可以用这个函数代替内置浮点函数:

v4.append(str_to_float(fields[3]))