这一节中所有的代码,如果不在QGIS命令行中执行,都需要导入如下包库
1from qgis.core import (
2 QgsRasterLayer,
3 QgsProject,
4 QgsPointXY,
5 QgsRaster,
6 QgsRasterShader,
7 QgsColorRampShader,
8 QgsSingleBandPseudoColorRenderer,
9 QgsSingleBandColorDataRenderer,
10 QgsSingleBandGrayRenderer,
11)
12
13from qgis.PyQt.QtGui import (
14 QColor,
15)
5. Using Raster Layers
5.1. Layer Details
A raster layer consists of one or more raster bands --- referred to as single band and multi band rasters. One band represents a matrix of values. A color image (e.g. aerial photo) is a raster consisting of red, blue and green bands. Single band rasters typically represent either continuous variables (e.g. elevation) or discrete variables (e.g. land use). In some cases, a raster layer comes with a palette and the raster values refer to the colors stored in the palette.
The following code assumes rlayer
is a
QgsRasterLayer
object.
rlayer = QgsProject.instance().mapLayersByName('srtm')[0]
# get the resolution of the raster in layer unit
print(rlayer.width(), rlayer.height())
919 619
# get the extent of the layer as QgsRectangle
print(rlayer.extent())
<QgsRectangle: 20.06856808199999875 -34.27001076999999896, 20.83945284300000012 -33.75077500700000144>
# get the extent of the layer as Strings
print(rlayer.extent().toString())
20.0685680819999988,-34.2700107699999990 : 20.8394528430000001,-33.7507750070000014
# get the raster type: 0 = GrayOrUndefined (single band), 1 = Palette (single band), 2 = Multiband
print(rlayer.rasterType())
0
# get the total band count of the raster
print(rlayer.bandCount())
1
# get the first band name of the raster
print(rlayer.bandName(1))
Band 1: Height
# get all the available metadata as a QgsLayerMetadata object
print(rlayer.metadata())
<qgis._core.QgsLayerMetadata object at 0x13711d558>
5.2. Renderer
When a raster layer is loaded, it gets a default renderer based on its type. It can be altered either in the layer properties or programmatically.
To query the current renderer:
print(rlayer.renderer())
<qgis._core.QgsSingleBandGrayRenderer object at 0x7f471c1da8a0>
print(rlayer.renderer().type())
singlebandgray
To set a renderer, use the setRenderer()
method of QgsRasterLayer
. There are a
number of renderer classes (derived from QgsRasterRenderer
):
Single band raster layers can be drawn either in gray colors (low values = black, high values = white) or with a pseudocolor algorithm that assigns colors to the values. Single band rasters with a palette can also be drawn using the palette. Multiband layers are typically drawn by mapping the bands to RGB colors. Another possibility is to use just one band for drawing.
5.2.1. Single Band Rasters
Let's say we want a render single band raster layer with colors ranging from
green to yellow (corresponding to pixel values from 0 to 255).
In the first stage we will prepare a
QgsRasterShader
object and configure
its shader function:
1fcn = QgsColorRampShader()
2fcn.setColorRampType(QgsColorRampShader.Interpolated)
3lst = [ QgsColorRampShader.ColorRampItem(0, QColor(0,255,0)),
4 QgsColorRampShader.ColorRampItem(255, QColor(255,255,0)) ]
5fcn.setColorRampItemList(lst)
6shader = QgsRasterShader()
7shader.setRasterShaderFunction(fcn)
The shader maps the colors as specified by its color map. The color map is provided as a list of pixel values with associated colors. There are three modes of interpolation:
linear (
Interpolated
): the color is linearly interpolated from the color map entries above and below the pixel valuediscrete (
Discrete
): the color is taken from the closest color map entry with equal or higher valueexact (
Exact
): the color is not interpolated, only pixels with values equal to color map entries will be drawn
In the second step we will associate this shader with the raster layer:
renderer = QgsSingleBandPseudoColorRenderer(rlayer.dataProvider(), 1, shader)
rlayer.setRenderer(renderer)
The number 1
in the code above is the band number (raster bands are
indexed from one).
Finally we have to use the
triggerRepaint()
method
to see the results:
rlayer.triggerRepaint()
5.2.2. Multi Band Rasters
By default, QGIS maps the first three bands to red, green and blue to
create a color image (this is the MultiBandColor
drawing style.
In some cases you might want to override these setting.
The following code interchanges red band (1) and green band (2):
rlayer_multi = QgsProject.instance().mapLayersByName('multiband')[0]
rlayer_multi.renderer().setGreenBand(1)
rlayer_multi.renderer().setRedBand(2)
In case only one band is necessary for visualization of the raster, single band drawing can be chosen, either gray levels or pseudocolor.
We have to use triggerRepaint()
to update the map and see the result:
rlayer_multi.triggerRepaint()
5.3. Query Values
Raster values can be queried using the
sample()
method of
the QgsRasterDataProvider
class.
You have to specify a QgsPointXY
and the band number of the raster layer you want to query. The method returns a
tuple with the value and True
or False
depending on the results:
val, res = rlayer.dataProvider().sample(QgsPointXY(20.50, -34), 1)
Another method to query raster values is using the identify()
method that returns a
QgsRasterIdentifyResult
object.
ident = rlayer.dataProvider().identify(QgsPointXY(20.5, -34), QgsRaster.IdentifyFormatValue)
if ident.isValid():
print(ident.results())
{1: 323.0}
In this case, the results()
method returns a dictionary, with band indices as keys, and band values as
values.
For instance, something like {1: 323.0}