arcpy制作脚本工具相关（制作并添加脚本工具、脚本工具消息提示、arcpy工具自动获取数据）

news2024/10/1 15:22:18

前言

采用arcpy制作arcgis的脚本工具，记录如下。

一、脚本工具的制作

制作脚本工具分为两步：

1、先编写对应功能的python脚步
2、将对应功能的脚本修改为，能自己手动选择输入输出数据的脚本
3、将修改完的脚本添加到工具箱

1、编写代码

我这个实例是用来提取特定高程的等高线，源代码如下：

# -*- coding: UTF-8 -*-
"""
@Date   :2023/6/25
"""
import time

"""
1、先写一段脚本
2、对脚本进行改写
"""
import os
import arcpy

print '程序开始：' + str(time.ctime())
# 设置工作空间
arcpy.env.workspace = r'E:\arcpy_study\Arcpy实例教程（数据）_小景空间_原始\6Arcpy脚本工具的制作-提取特殊等高线'

# 输入要素
inFeatures = 'coutour.shp'

# 添加一个字段，用来标记高程是否可以整除10
filedName = "Mark"  # 字段名称
filedPrecision = 2  # 字段长度
filedAlias = "整除50标记"  # 字段别名  能整除50标记为1，不能整除50标记为0

# 列出所有字段
filedObject = arcpy.ListFields(inFeatures)
# 得到所有字段名，如果mark字段存在的话，就删除该字段
filedNameList = []
for filed in filedObject:
    # filed.required：字段为必填项时为ture
    if not filed.required:
        filedNameList.append(filed.name)
print filedNameList
if filedName in filedNameList:
    arcpy.DeleteField_management(inFeatures, filedName)
    print '删除已有的mark字段'

# 添加字段
arcpy.AddField_management(inFeatures, filedName, "LONG", filedPrecision, "", "", filedAlias, "NULLABLE")

filed1 = "Elev"
filed2 = "Mark"

# 更新查询(为每一条记录做标记)
cursor = arcpy.UpdateCursor(inFeatures)
for row in cursor:
    if((int)(row.getValue(filed1))%50 == 0):
        row.setValue(filed2, 1)
    else:
        row.setValue(filed2, 0)
    cursor.updateRow(row)
print '程序结束：' + str(time.ctime())

在原有字段的基础上，增加一个Mark字段，用来标识这条等高线是否能被50整除，若能被50整除，则标识为1，否则为0.

在这个例子中，有一个输入要素，标识要操作的要素名称。

2、修改脚本

在源代码中，我将要修改的要素名称写死了，实际情况肯定不行，所有我要修改为自己可以手动选择要素，工作空间也不需要了。修改后的代码如下：

# -*- coding: UTF-8 -*-
"""
@Date   :2023/6/26
"""
import time
import os
import arcpy

print '程序开始：' + str(time.ctime())
# 设置工作空间
# arcpy.env.workspace = r'E:\arcpy_study\Arcpy实例教程（数据）_小景空间_原始\6Arcpy脚本工具的制作-提取特殊等高线'

# 输入要素
# inFeatures = 'coutour.shp'
# 自己选择输入要素
inFeatures = arcpy.GetParameterAsText(0)  # 获取第一个输入要素的文本

# 添加一个字段，用来标记高程是否可以整除10
filedName = "Mark"  # 字段名称
filedPrecision = 2  # 字段长度
filedAlias = "整除50标记"  # 字段别名  能整除50标记为1，不能整除50标记为0

# 列出所有字段
filedObject = arcpy.ListFields(inFeatures)
# 得到所有字段名，如果mark字段存在的话，就删除该字段
filedNameList = []
for filed in filedObject:
    # filed.required：字段为必填项时为ture
    if not filed.required:
        filedNameList.append(filed.name)
print filedNameList
if filedName in filedNameList:
    arcpy.DeleteField_management(inFeatures, filedName)
    print '删除已有的mark字段'

# 添加字段
arcpy.AddField_management(inFeatures, filedName, "LONG", filedPrecision, "", "", filedAlias, "NULLABLE")

filed1 = "Elev"
filed2 = "Mark"

# 更新查询(为每一条记录做标记)
cursor = arcpy.UpdateCursor(inFeatures)
for row in cursor:
    if ((int)(row.getValue(filed1)) % 50 == 0):
        row.setValue(filed2, 1)
    else:
        row.setValue(filed2, 0)
    cursor.updateRow(row)
print '程序结束：' + str(time.ctime())