python GDAL库的简单使用

1 打开数据集

file_path = "./test.hdf"
ori_dataset = gdal.Open(file_path)  # 返回一个gdal.Dataset类型的对象
sub_datasets = ori_dataset.GetSubDatasets()     # 对于hdf类型的文件需要这一步来获取子数据集
# tiff等文件格式则不需要再打开子数据集，直接对ori_dataset操作即可
dataset = gdal.Open(sub_datasets[0][0])     # 打开hdf中的子数据集

2 获取图像信息

num_of_band = dataset.RasterCount   # 波段数
cols = dataset.RasterXSize  # 图像长度
rows = dataset.RasterYSize  # 图像宽度
proj = dataset.GetProjection()  # 获取投影,返回投影的字符串

3 读取数据到numpy数组

band1 = dataset.GetRasterBand(1)    # 获取第一波段，从1开始
band1_arr = band1.ReadAsArray(0, 0, cols, rows)     # 读取0-rows行， 0-cols列的数据，返回numpy数组
band2_arr = dataset.GetRasterBand(2).ReadAsArray(0, 0, cols, rows)

#也可以不分波段读取，直接全部读进numpy数组
im_data = dataset.ReadAsArray(0, 0, cols, rows)
band1 = im_data[0, :, :]