使用 python 處理 nc 數據
前言
這兩天幫一個朋友處理了些 nc 數據,本以為很簡單的事情,沒想到裏面涉及到了很多的細節和坑,無論是“知難行易”還是“知易行難”都不能充分的說明問題,還是“知行合一”來的更靠譜些,既要知道理論又要知道如何實現,於是經過不太充分的研究後總結成此文,以記錄如何使用 python 處理 nc 數據。
一、nc 數據介紹
nc 全稱 netCDF(The Network Common Data Form),可以用來存儲一系列的數組,就是這麽簡單(參考https://www.unidata.ucar.edu/software/netcdf/docs/netcdf_introduction.html)。
既然 nc 可以用來一系列的數組,所以經常被用來存儲科學觀測數據,最好還是長時間序列的。
試想一下一個科學家每隔一分鐘采集一次實驗數據並存儲了下來,如果不用這種格式存儲,時間長了可能就需要創建一系列的 csv 或者 txt 等,而采用 nc 一個文件就可以搞定,是不是很方便。
更方便的是如果這個科學實驗與氣象、水文、溫度等地理信息稍微沾點邊的,完全也可以用 nc 進行存儲, GeoTiff 頂多能多存幾個波段(此處波段可以認為是氣象、水文等不同信號),而 nc 可以存儲不同波段的長時間觀測結果,是不是非常方便。
可以使用 gdal 查看數據信息,執行:
gdalinfo name.nc
即可得到如下信息:
Driver: netCDF/Network Common Data Format
Files : test.nc
Size is 512, 512
Coordinate System is `'
Subdatasets:
SUBDATASET_1_NAME=NETCDF:"test.nc":T2
SUBDATASET_1_DESC=[696x130x120] T2 (32-bit floating-point)
SUBDATASET_2_NAME=NETCDF:"test.nc":PSFC
SUBDATASET_2_DESC=[696x130x120] PSFC (32-bit floating-point)
SUBDATASET_3_NAME=NETCDF:"test.nc":Q2
SUBDATASET_3_DESC=[696x130x120] Q2 (32-bit floating-point)
SUBDATASET_4_NAME=NETCDF:"test.nc":U10
SUBDATASET_4_DESC=[696x130x120] U10 (32-bit floating-point)
SUBDATASET_5_NAME=NETCDF:"test.nc":V10
SUBDATASET_5_DESC=[696x130x120] V10 (32-bit floating-point)
SUBDATASET_6_NAME=NETCDF:"test.nc":RAINC
SUBDATASET_6_DESC=[696x130x120] RAINC (32-bit floating-point)
SUBDATASET_7_NAME=NETCDF:"test.nc":SWDOWN
SUBDATASET_7_DESC=[696x130x120] SWDOWN (32-bit floating-point)
SUBDATASET_8_NAME=NETCDF:"test.nc":GLW
SUBDATASET_8_DESC=[696x130x120] GLW (32-bit floating-point)
SUBDATASET_9_NAME=NETCDF:"test.nc":LAT
SUBDATASET_9_DESC=[130x120] LAT (32-bit floating-point)
SUBDATASET_10_NAME=NETCDF:"test.nc":LONG
SUBDATASET_10_DESC=[130x120] LONG (32-bit floating-point)
Corner Coordinates:
Upper Left ( 0.0, 0.0)
Lower Left ( 0.0, 512.0)
Upper Right ( 512.0, 0.0)
Lower Right ( 512.0, 512.0)
Center ( 256.0, 256.0)
每一個 SUBDATASET 表示記錄的是一種格式的數據(氣象、水文等等),如果要想查看此 SUBDATASET 的具體信息,可以執行:
gdalinfo NETCDF:name.nc:SUBDATASET_NAME
此處的 SUBDATASET_NAME 為上面的 T2、PSFC 等等,可以得到如下信息:
Driver: netCDF/Network Common Data Format
Files: test.nc
Size is 120, 130
Coordinate System is `'
Metadata:
LAT#description=LATITUDE, SOUTH IS NEGATIVE
LAT#FieldType=104
LAT#MemoryOrder=XY
LAT#stagger=
LAT#units=degree_north
Corner Coordinates:
Upper Left ( 0.0, 0.0)
Lower Left ( 0.0, 130.0)
Upper Right ( 120.0, 0.0)
Lower Right ( 120.0, 130.0)
Center ( 60.0, 65.0)
Band 1 Block=120x1 Type=Float32, ColorInterp=Undefined
NoData Value=9.96920996838686905e+36
Unit Type: degree_north
Metadata:
description=LATITUDE, SOUTH IS NEGATIVE
FieldType=104
MemoryOrder=XY
NETCDF_VARNAME=LAT
stagger=
units=degree_north
此處只有一個 Band ,每一個 Band 記錄了一個時間點(或者其他區分形式)的一條記錄,這個記錄是一個數組。
所以看到這裏,各位應該已經明白了,可以直接使用 GDAL 處理 nc 數據,比如直接使用 gdalwarp 將某個 SUBDATASET 轉成 GeoTiff 等等,此處暫且不表,各位只需要查閱一下 gdalwarp 手冊即可知道如何處理。
明白了以上信息基本也就清楚了如何處理此數據。
二、數據處理
python 是運用非常廣泛,自然其下各種類庫非常豐富,專業一點的說法就叫生態豐富。
2.1 netCDF4
此框架可以直接將 nc 讀取成數組(詳細信息參考https://github.com/Unidata/netcdf4-python)。讀取方式如下:
dataset = netCDF4.Dataset('name.nc') # open the dataset
這樣即可讀出整個 nc 中的數據信息,如果需要獲取某個 SUBDATASET 只需要使用 dataset[SUBDATASET_NAME]
即可,返回的是一個三維數組,表示不同時間段(或其他區分方式下)的數據信息。
我們可以對此數組做各種操作,如求平均值、方差等等,又讓我想起了大學裏的那一堆枯燥但又讓人很有興趣的實驗課程。當然,此處如果使用 numpy 框架進行處理,會起到事半功倍的效果,如求長時間序列下的平均值:
np_arr = np.asarray(dataset[SUBDATASET_NAME])
average_arr = np.average(np_arr, axis=0)
到這裏跟地信有關的同誌都會看出一個問題,此框架只能對數據進行處理,而不能進行與位置有關的操作,這就導致數據無法變成直白的地圖可視化效果。其實任何數據都是相通的,我們可以采用此種方式處理完後轉為 GeoTiff 等,當然我們也可以直接采用 GeoTiff 的處理流程來進行處理。
2.2 rasterio
rasterio 是 Mapbox 開源的空間數據處理框架,功能非常強大,此處不細說,只表如何處理我們的 nc 數據。
當然第一種方式就是使用 netCDF4 處理完之後,使用此框架寫入 GeoTiff,但是這樣不太優雅,而且使用了兩個框架,明顯過於麻煩,我們直接使用此框架從讀數據開始處理。
此處讀的時候就有技巧了,要像采用 gdalinfo 讀取 SUBDATASET 一樣來直接讀取此 SUBDATASET 數據,如下:
with rio.open('NETCDF:name.nc:SUBDATASET_NAME') as src:
print(src.meta)
dim = int(src.meta['count'])
src.read(range(1, dim + 1))
即給 open 函數傳入 NETCDF:name.nc:SUBDATASET_NAME
,采用 src.read(range(1, dim + 1))
可以直接讀出此範圍內所有 Band (時間點)的信息,範圍可以自己設定,註意從 0 開始,當然也可以僅讀取某個 Band 的信息。
src.meta 記錄了此 SUBDATASET 的元數據信息,與 gdalinfo 看到的基本相同。
這樣我們就可以繼續將此數據使用 numpy 等框架進行處理,處理完之後更重要的是要寫入 GeoTiff 中(直白的說就是添加空間信息)。
也很簡單,如下即可:
with rio.open(newfile, 'w', **out_meta) as dst:
dst.write_band(1, res_arr)
newfile 為存儲路徑,res_arr 為計算結果數組,註意尺寸不要發生變化(width*height),out_meta 為目標文件的元數據描述信息,可以直接將上面 src.meta 進行簡單處理即可。
out_meta =
meta.update({"driver": "GTiff",
"dtype": "float32",
'count': 1,
'crs': 'Proj4: +proj=longlat +datum=WGS84 +no_defs',
'transform': rasterio.transform.from_bounds(west, south, east, north, width, height)
})
crs 表示目標數據空間投影信息,transform 表示目標文件 空間範圍信息,可以通過經緯度信息和圖像尺寸等計算得到。
dst.write_band
將數據寫入對應波段,當然此處也可以寫入多個波段,根據計算結果而定,同樣從 1 開始。
三、總結
本文簡單介紹了 nc 數據的特點及如何使用 python 處理 nc 數據。每個目標都有多條路可以達到,重要的是找到那條自己喜歡的和適合自己的路,然而話又說回來,即使走的不是想要的那條路,不是一樣可以達到目標嘛!所以關鍵是要找到自己的目標。
使用 python 處理 nc 數據