基于python的批处理研究区代码

117 阅读1分钟

好久没来掘金写文章了,最近在忙着写论文,我分享一下我用python批处理的研究区吧 这个python需要安装arcpy,只需要修改你的输入文件目录和输出文件目录以及掩膜Shapefile的路径即可。

# -*- coding: utf-8 -*-
import arcpy
import os

# 设置环境
arcpy.env.workspace = r"E:\jiangshui\2020"  # 输入文件目录
output_folder = r"E:\jiangshui\2020pichuli"  # 输出文件目录
mask_shp = r"D:\bianjie\XJ_bj\XJ_BJ3.shp"  # 掩膜Shapefile路径,确保这里的路径和文件名完全正确

# 检查输出目录是否存在,如果不存在则创建
if not os.path.exists(output_folder):
    os.makedirs(output_folder)

# 获取所有的TIF文件
tif_files = arcpy.ListRasters("*", "TIF")

# 对每个TIF文件进行掩膜裁剪
for tif in tif_files:
    print("Processing {}...".format(tif))
    # 设置输出文件的路径
    output_path = os.path.join(output_folder, tif)
    # 执行裁剪操作
    try:
        arcpy.gp.ExtractByMask_sa(tif, mask_shp, output_path)
    except Exception as e:
        print(e)
        arcpy.AddError(e)

print("All files have been processed.")

这几天每天都会发一篇我在写论文用到python进行提取数据的文章,有用python中arcpy的小伙伴可以评论或者私信我。