Python下遥感图像文件的读写

python学习记录

图像文件的读写

记录下自己的一点学习

读取Geotiff

from osgeo import gdal
import numpy as np
dataset=gdal.Open("fdem.tif") #打开文件
im_width = dataset.RasterXSize #栅格矩阵的列数
im_height = dataset.RasterYSize #栅格矩阵的行数
im_bands = dataset.RasterCount #波段数
im_geotrans = dataset.GetGeoTransform() #仿射矩阵，左上角像素的大地坐标和像素分辨率
im_proj = dataset.GetProjection() #地图投影信息，字符串表示
im_data = dataset.ReadAsArray(0,0,im_width,im_height)
type(im_data),im_data.shape
del dataset
#显示栅格数据
print im_data[0, 10:15,20:25] #查看波段 0,10~14 行和 20~25 列的数据
print im_geotrans

显示栅格的时候有问题too many indices for array，实验用的数据是单波段因此删掉print im_data[0, 10:15,20:25] 中的波段0可以正常运行。

#判断栅格数据的数据类型
if 'int8' in im_data.dtype.name:
 datatype = gdal.GDT_Byte
elif 'int16' in im_data.dtype.name:
 datatype = gdal.GDT_UInt16
else:
 datatype = gdal.GDT_Float32
#判读数组维数
if len(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")
dataset = driver.Create("fdem_new.tif", im_width, im_height, im_bands, datatype)
dataset.SetGeoTransform(im_geotrans) #写入仿射变换参数
dataset.SetProjection(im_proj) #写入投影
if im_bands == 1:
 dataset.GetRasterBand(1).WriteArray(im_data) #写入数组数据
else:
 for i in range(im_bands):
 dataset.GetRasterBand(i+1).WriteArray(im_data[i])
del dataset 

栅格数据主要是栅格矩阵（像元值，像元大小，波段数）和大地坐标（空间参考，仿射变换）表现