QgsRasterLayer转ndarray

最新推荐文章于 2023-10-20 12:09:02 发布

欧阳罢笔

最新推荐文章于 2023-10-20 12:09:02 发布

阅读量640

点赞数

分类专栏： QGIS Python paddle 文章标签： python qgis numpy

本文链接：https://blog.csdn.net/qq_39798423/article/details/121095383

版权

Python 同时被 3 个专栏收录

14 篇文章 2 订阅

订阅专栏

paddle

13 篇文章 2 订阅

订阅专栏

QGIS

6 篇文章 2 订阅

订阅专栏

QgsRasterLayer转ndarray

目前在QGIS中没看到读取到栅格后怎么转化为nd数组在后台进行一些操作（比如跑个深度学习模型啥的）。于是经过参考和探索，先出了一个折中的办法，将栅格转为nd数组，效率不高，暂时保留，后续发现更好的方法再来更新。代码如下：

import numpy as np


def raster2ndarray(lyr):
    '''
        input: lyr(QgsMapLayerType.QgsRasterLayer)
        output: _(ndarray)
    '''
    provider= lyr.dataProvider()
    # 多少个波段
    blocks = []
    for c in range(lyr.bandCount()):
        blocks.append(provider.block((c + 1), lyr.extent(), lyr.width(), lyr.height()))
    values=[]
    for i in range(lyr.width()):
        tmps = []
        for j in range(lyr.height()):
            tmps.append([blocks[k].value(i, j) for k in range(lyr.bandCount())])
        values.append(tmps)
    return np.array(values)