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
#判读数组维数