Просмотр кода страницы Замена значений в растрах с помощью GDAL/Numpy

{{Статья|Опубликована|remap-gdal}}
{{Аннотация|Описание и скрипт.}}

Задача: заменить в растре одно значение на другое.

Для решения можно использовать возможности GDAL для чтения/записи растров и Numpy для операций замены. Данный способ использовался, например, для нормализации ошибочных значений NODATA в данных [http://gis-lab.info/qa/srtm.html SRTM 5x5].

=== Реализация ===

Все основные моменты работы с растрами изложены в статье «[http://gis-lab.info/qa/gdal-python.html Работа с растрами при помощи GDAL и Python]». Остановимся на ключевых деталях реализации:

Так как мы хотим только поменять значение, то важно сохранить растр в том же числовом формате, что и исходный. Для этого нужно при чтении получить исходный числовой формат:

<syntaxhighlight lang="python">
gd = gdal.Open( input )
intype = gd.GetRasterBand( 1 ).DataType
</syntaxhighlight>

и использовать его же при сохранении результирующего растра:

<syntaxhighlight lang="python">outRaster = driver.Create( output, xsize, ysize, 1, intype )</syntaxhighlight>

Также полезным может оказаться установка значения NoData, если для его установки и заменяются значения:

<syntaxhighlight lang="python">outRaster.GetRasterBand( 1 ).SetNoDataValue(-32678)</syntaxhighlight>

Основная работа производится с помощью функций обработки массивов Numpy, значительно более быстрых чем попиксельная обработка. Цикл поиска-замены значения выглядит следующим образом:

<syntaxhighlight lang="python">
temp1_bool = numpy.equal(raster,inval)
numpy.putmask(raster,temp1_bool,outval)
</syntaxhighlight>

=== Скрипт ===

Запуск скрипта производится следующим образом:

<pre>remap.py input_raster output_raster input_value output_value</pre>

В качестве аргументов используется имя исходного и конечного растра, а также исходное и конечное значение. Например, для замены значения 255 на -32678:

<pre>remap.py input.tif output.tif 255 -32678</pre>

[http://gis-lab.info/programs/python/remap.7z Скачать] скрипт.

'''Предупреждение''': на данный момент описанная реализация относительно медленная, операции на больших растрах могут занимать значительное время.

=== Ссылки по теме ===
* [http://gis-lab.info/qa/gdal-python.html Работа с растрами при помощи GDAL и Python]
* [http://gis-lab.info/qa/ogr-python.html Работа с векторными данными при помощи OGR и Python]