Невозможно написать растр, используя Gdal в Python. Ошибка: dict 'объект не имеет атрибута' shape ' - PullRequest
Новая
       10

Невозможно написать растр, используя Gdal в Python. Ошибка: dict 'объект не имеет атрибута' shape '

1 голос
/ 31 января 2020

Я пытаюсь создать растровый файл после заполнения NO DATA некоторым значением, используя gdal в Python.

У меня есть функция, которая возвращает мне растровый массив. 

def raster2array(rasterfn):
try:
    bndNum_Val_Dic={}
    raster = gdal.Open(rasterfn)
    for bandNum in range(raster.RasterCount):
        bandNum += 1
        band=raster.GetRasterBand(bandNum)
        bandVal=band.ReadAsArray()
        bndNum_Val_Dic[bandNum]=bandVal
    raster=None
    return bndNum_Val_Dic
except Exception as e:
    print(e)

Используя массив, сгенерированный из этой функции, я пытаюсь написать свой растр, который выдает ошибку "outband.WriteArray (array)", что dict "объект не имеет атрибута" shape ". 

import numpy as np
import gdal
from osgeo import osr
rasterfn ="MAH_20.tif"
newRasterfn ="MAH_FND.tif"
array= raster2array(rasterfn)
newValue = 100
Driver= 'GTiff'
bandNumber=1
raster = gdal.Open(rasterfn)
geotransform = raster.GetGeoTransform()
originX = geotransform[0]
originY = geotransform[3]
pixelWidth = geotransform[1]
pixelHeight = geotransform[5]
cols = raster.RasterXSize
rows = raster.RasterYSize
bandCount=raster.RasterCount
rasterDataType=raster.GetRasterBand(bandNumber).DataType
global Flag
if(Flag):
    driver = gdal.GetDriverByName(Driver)
    global outRaster
    outRaster = driver.Create(newRasterfn, cols, rows, bandCount, rasterDataType)
    Flag=False

outband = outRaster.GetRasterBand(bandNumber)
outRaster.SetGeoTransform((originX, pixelWidth, 0, originY, 0, pixelHeight))
outband = outRaster.GetRasterBand(bandNumber)
outband.WriteArray(array)
outRasterSRS = osr.SpatialReference()
outRasterSRS.ImportFromWkt(raster.GetProjectionRef())
outRaster.SetProjection(outRasterSRS.ExportToWkt())
outRaster.GetRasterBand(bandNumber).SetNoDataValue(newValue)
raster=None
if(bandNumber==bandCount):
    outRaster=None
    outband=None
    raster=None

I Я использую python 3.5 и GDAL 3.0.2. Есть ли способ это исправить? Любая помощь будет оценена

1 Ответ

0 голосов
/ 31 января 2020

Вы пытаетесь два написать словарь, а GDAL ожидает массив Numpy. Не совсем ясно, какие данные вы пытаетесь записать, но изменение вашего оператора записи на что-то, как показано ниже, должно по крайней мере избавиться от сообщения об ошибке. Но убедитесь, что вы пишете правильную группу. 

outband.WriteArray(array[bandNumber])
...