/
gdaldiy.py
47 lines (43 loc) · 1.68 KB
/
gdaldiy.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# -*- coding: utf-8 -*-
"""
Created on Fri Oct 25 20:57:37 2019
@author: Neoooli
"""
from osgeo import gdal
import numpy as np
list1 = ["byte","uint8","uint16","int16","uint32","int32","float32","float64","cint16","cint32","cfloat32","cfloat64"]
list2 = [gdal.GDT_Byte,gdal.GDT_Byte,gdal.GDT_UInt16,gdal.GDT_Int16,gdal.GDT_UInt32,gdal.GDT_Int32,gdal.GDT_Float32,gdal.GDT_Float64,gdal.GDT_CInt16,gdal.GDT_CInt32,gdal.GDT_CFloat32,gdal.GDT_CFloat64]
def imgread(path):
img = gdal.Open(path)
# col = img.RasterXSize #col
# row = img.RasterYSize #row
# img_arr = img.ReadAsArray(0,0,col,row) #设置读取图像的范围(宽/高)
c = img.RasterCount
img_arr = img.ReadAsArray() #读取整幅图像
if c>1:
img_arr = img_arr.swapaxes(1,0)
img_arr = img_arr.swapaxes(2,1)
del img
return img_arr
def imgwrite(path,narray,compress="None"):
s=narray.shape
dt_name=narray.dtype.name
for i in range(len(list1)):
if list1[i] in dt_name.lower():
datatype=list2[i]
break
else:
datatype=list2[0]
if len(s)==2:
row,col,c=s[0],s[1],1
driver = gdal.GetDriverByName('GTiff')
dataset = driver.Create(path,col,row,c,datatype,options=["COMPRESS="+compress]) #GDAL存储为宽/长/波段->列/行/通道
dataset.GetRasterBand(1).WriteArray(narray)
del dataset
elif len(s)==3:
row,col,c = s[0], s[1], s[2]
driver = gdal.GetDriverByName('GTiff')
dataset = driver.Create(path,col,row, c, datatype)
for i in range(c):
dataset.GetRasterBand(i + 1).WriteArray(narray[:,:,i])
del dataset