Python疊加兩幅柵格圖像

目的
現有兩幅柵格圖像,一個是某地區道路柵格圖,一個是某地區土地利用類型圖,需要將道路疊加到土地利用類型圖中,即疊加後,重合的像元值以道路圖爲準,其餘的像元值仍是土地利用類型圖原有的像元值。
圖1 道路信息圖
這裏寫圖片描述
圖2 土地利用類型圖
這裏寫圖片描述
圖3 結果圖
這裏寫圖片描述
具體實現

from gdalconst import *
from osgeo import gdal
import osr
import sys
import copy

#疊加兩個柵格圖像(一個道路柵格圖,一個土地利用類型圖),兩幅圖像重疊的像元值都是第一個圖像的值,
#未重疊的像元值還是土地利用類型圖上的值,最終結果便是土地利用類型圖上面多了道路信息。

roadFile =  'E:\\Exercise\\test\\grasstest\\road_rastercalc.tif'
landuseFile =  'E:\\Exercise\\test\\grasstest\\landuse.tif'
roadDs = gdal.Open(roadFile, GA_ReadOnly)
landuseDs = gdal.Open(landuseFile, GA_ReadOnly)
if roadDs is None:
    print 'Can not open ', roadFile
    sys.exit(1)

geotransform = roadDs.GetGeoTransform()
projection=roadDs.GetProjection()
cols = roadDs.RasterXSize
rows = roadDs.RasterYSize
roadBand = roadDs.GetRasterBand(1)
roadData = roadBand.ReadAsArray(0,0,cols,rows)
roadNoData = roadBand.GetNoDataValue()

landuseBand = landuseDs.GetRasterBand(1)
landuseData = landuseBand.ReadAsArray(0,0,cols,rows)
landuseNoData = landuseBand.GetNoDataValue()


result = landuseData

for i in range(0,rows):
    for j in range(0,cols):
        if(abs(roadData[i,j] - 20) < 0.0001):
           result[i,j] = 20
        if((abs(landuseData[i,j] - landuseNoData)>0.0001) and (abs(roadData[i,j] - roadNoData) < 0.0001)):
            result[i,j] = landuseData[i,j]
        if((abs(landuseData[i,j] - landuseNoData)<0.0001) and (abs(roadData[i,j] - roadNoData) < 0.0001)):
            result[i,j] = landuseNoData
#write result to disk
resultPath = 'E:\\Exercise\\test\\grasstest\\result_landuse.tif'

format = "GTiff"   
driver = gdal.GetDriverByName(format)
ds = driver.Create(resultPath, cols, rows, 1, GDT_Float32)
ds.SetGeoTransform(geotransform)
ds.SetProjection(projection)
ds.GetRasterBand(1).SetNoDataValue(landuseNoData)
ds.GetRasterBand(1).WriteArray(result)    
ds = None

print 'ok---------'
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章