使用配色方案保存GeoTiff

时间:2014-06-26 16:10:51

标签: arrays numpy geotiff

我正在尝试将numpy数组输出保存为GeoTiff,并且代码运行最成功,但输出图像的数据具有棕褐色调的比例(而不是像我的代码生成的正常颜色方案图像)和黑色背景(而不是我的代码为图像生成的白色/无数据背景)。

这里是将我的数组保存到GeoTiff的代码。我可以在某处添加一行关于使no-data = 0,并使数据方案变为彩色吗?

from osgeo import gdal, osr, ogr, os 
from gdalconst import *  

def array2raster(newRasterfn,rasterOrigin,pixelWidth,pixelHeight,array):

    cols = array.shape[1]
    rows = array.shape[0]
    originX = rasterOrigin[0]
    originY = rasterOrigin[1]

    driver = gdal.GetDriverByName( 'GTiff' )
    outRaster = driver.Create(newRasterfn, cols, rows, 1, gdal.GDT_Float32)
    outRaster.SetGeoTransform((originX, pixelWidth, 0, originY, 0, pixelHeight))
    outband = outRaster.GetRasterBand(1)
    outband.WriteArray(array)
    # outRaster = driver.Create( 'CORDC_GTIFF/working_CA.tiff', 300, 300, 1,     gdal.GDT_Int32)
    proj = osr.SpatialReference()  
    proj.ImportFromEPSG(4326) 
    outRaster.SetProjection(proj.ExportToWkt())  
    # geotransform = (1,0.1,0,40,0,0.1)  


rasterOrigin = (-127,42)
pixelWidth = .01
pixelHeight = .01
newRasterfn = 'CORDC_GTIFF/cordc_working_CA.tif'
array = np.array(spd)


reversed_arr = array[::-1] # reverse array so the tif looks like the array
array2raster(newRasterfn,rasterOrigin,pixelWidth,pixelHeight,reversed_arr) # convert array to raster

1 个答案:

答案 0 :(得分:1)

您可以使用乐队的SetNoDataValue方法设置无数据值:

outband = outRaster.GetRasterBand(1)
outband.SetNoDataValue(0)
outband.WriteArray(array)

与无数据值匹配的区域应显示为透明