python常见数据读写

定义数组

import numpy as np
tankongjihe = np.zeros((13,1500, 8), float)

写入数组到csv

import numpy as np
a = np.array([1,2,3])
np.savetxt('test000002.csv', a, delimiter=',')

csv读取数组

import numpy as np
data = np.loadtxt('test000002.csv', delimiter=',')
#这种不能读取三维的

import numpy as np
np.save('tankongjihe',arr = tankongjihe)
np.load('tankongjihe.npy')
#这种可以写入读取三维的数据，但是不能在表格查看

将数组存为netcdf文件

shf_analysis_3D = shf_analysis.reshape(data.sshf.shape)
#这是需要写进nc的数据

def write_to_nc(data,file_name_path):
    import netCDF4 as nc
    lonS=np.linspace(0,359.8,1440)
    latS=np.linspace(90,-90,721)
    da=nc.Dataset(file_name_path,'w',format='NETCDF4')
    da.createDimension('lon',1440) #创建坐标点
    da.createDimension('lat',721) #创建坐标点
    da.createVariable("lon",'f',("lon")) #添加coordinates 'f'为数据类型，不可或缺
    da.createVariable("lat",'f',("lat")) #添加coordinates 'f'为数据类型，不可或缺
    da.variables['lat'][:]=latS  #填充数据
    da.variables['lon'][:]=lonS  #填充数据
    da.createVariable(write_to_nc.__code__.co_varnames[0],'f8',('lat','lon')) #创建变量，shape=(721,1440) 'f'为数据类型，不可或缺
    da.variables[write_to_nc.__code__.co_varnames[0]][:]=data  #填充数据 
    da.close()
os.remove('shf_analysis_3d.nc')
write_to_nc(shf_analysis_3D,'shf_analysis_3d.nc')

从netcdf 文件中读取数据

import xarray as xr
filename1 = 'flux_dp_test.nc'
data=xr.open_dataset(filename1)
u10 = data.u10.data