Python地理数据处理 二十:基于arcpy批量操作(二)

1. 栅格计算器(RasterCalculator_sa)

arcpy能实现批量化的栅格操作,提高工作效率,避免机械化的重复操作。
之前我也讲到过arcgis的栅格计算器操作,arcgis软件中也有批量操作功能,但是也比较繁琐,所以决定使用arcpy进行处理,大概30s就能完成上百幅图的栅格计算。

代码解释:
提取栅格图层中大于10的像元,并将其赋值为1,其余像元赋值为0。

# -*- coding: cp936 -*-
import arcpy 

arcpy.CheckOutExtension('Spatial')

# 设置工作环境
arcpy.env.workspace = r"D:\dataset"
# 获取所有栅格数据的列表
raster_list = arcpy.ListRasters("*", "TIF")

# 遍历栅格数据列表进行计算
for raster in raster_list:
    print raster
    # 构建计算表达式(注意这里的双引号!!!)
    expression = "Con("'"{0}"'" > 10, 1, 0)".format(raster)
    # 执行计算
    arcpy.gp.RasterCalculator_sa(expression, r"D:\dataset\output\{0}".format(raster))

2. 按属性提取要素(ExtractByAttributes_sa)

函数解释:
提取栅格属性表中“value” = 10的所有像元:

# -*- coding: cp936 -*-
import arcpy

arcpy.CheckOutExtension('Spatial')

# 设置工作环境
arcpy.env.workspace = r"D:\dataset"
# 获取所有栅格数据的列表
raster_list = arcpy.ListRasters("*", "TIF")

# 遍历栅格数据列表进行计算
for raster in raster_list:
    print raster
    out_raster = "D:/dataset/output/" + raster
    # 定义提取条件,提取属性表中 "Value" = 10 的像元
    where_clause = "Value = 10"
    # 执行栅格提取
    arcpy.gp.ExtractByAttributes_sa(raster, where_clause, out_raster)

3. 栅格数据转化为矢量数据(RasterToPolygon_conversion)

# -*- coding: cp936 -*-
import arcpy
import os

# 设置工作空间和输出路径
arcpy.env.workspace = r"D:\Datasets\tif"
out_path = r"D:\Datasets\shapefile"

# 遍历文件夹中的所有栅格数据
for root, dirs, files in os.walk(arcpy.env.workspace):
    for file in files:
        if file.endswith(".tif"):  # 如果是tif格式的栅格数据
            # 获取栅格数据路径和名称
            input_raster = os.path.join(root, file)

            # 设置输出要素类名称
            output_feature_class = os.path.splitext(file)[0] + ".shp"

            # 栅格转换为矢量
            arcpy.RasterToPolygon_conversion(input_raster, os.path.join(out_path, output_feature_class))

猜你喜欢

转载自blog.csdn.net/amyniez/article/details/129807332