Python有许多的库可以用来读写栅格数据,比如常见的GDAL、rasterio等,这里介绍一下如何使用GDAL来读写栅格数据,把读写栅格数据分别封装成了两个函数,直接复制使用即可:
读取栅格数据:
python
from osgeo import gdal
from osgeo import ogr
from osgeo import osr
import numpy as np
# read data: nodata -> 0
def ReadData(filepath): # 输入文件路径,tif格式
gdal.AllRegister()
ds = gdal.Open(filepath)
clos = ds.RasterXSize # 列数
rows = ds.RasterYSize # 行数
bands = ds.RasterCount # 波段数
geotrans = ds.GetGeoTransform() # 坐标系基准
proj = ds.GetProjection() # 投影
data = ds.ReadAsArray()
data[data == data[0,0]] = 0 # nodata转为0,这里可以自己设置
return [clos, rows, data, geotrans, proj]
写出栅格数据:
python
def Write2Tiff(newpath,im_data,im_geotrans,im_proj): # 输入写出路径;写出数据;写出数据的基准;投影
if 'int8' in im_data.dtype.name:
datatype = gdal.GDT_Int16
elif 'int16' in im_data.dtype.name:
datatype = gdal.GDT_Int16
else:
datatype = gdal.GDT_Float32
if len(im_data.shape) == 3:
im_bands, im_height, im_width = im_data.shape
else:
im_bands, (im_height, im_width) = 1, im_data.shape
driver = gdal.GetDriverByName('GTiff')
new_data = driver.Create(newpath, im_width, im_height, im_bands, datatype)
new_data.SetGeoTransform(im_geotrans)
new_data.SetProjection(im_proj)
if im_bands == 1:
new_data.GetRasterBand(1).WriteArray(im_data)
else:
for i in range(im_bands):
new_data.GetRasterBand(i+1).WriteArray(im_data[i])
del new_data
以上就是关于如何使用GDAL读写栅格数据。