Arcgis添加字段(Python接口)

缺乏、安全感 2022-09-09 02:23 244阅读 0赞

> 突然发现Arcgis对Python的支持非常的好，所以就试着实现了一个小功能。虽然这些功能也可以使用C\#+AE进行实现，但是有时候总感觉杀鸡用牛刀的感觉\*\_\*。

话不多说，毕竟python程序一般都比较短，所以就直接上代码了。

#
    #为shp文件添加指定字段
    #
    
    #-*-coding:utf-8-*-
    #导入所需要的包
    import sys
    reload(sys)
    sys.setdefaultencoding('utf-8')		#Arcgis对中文兼容不是很好，这一步是让其支持utf-8编码方式
    import arcpy
    import os
    
    
    #输入输出路径
    inputPath = arcpy.GetParameterAsText(0) 
    inputFields = arcpy.GetParameterAsText(1) 
    outputPath = arcpy.GetParameterAsText(2)
    
    #映射表
    classMap = {      #类别映射表
        "未标注\背景":0,
        "耕地":1,
        "园地":2,
    
        "WBZ":0,
        "GD":1,
        "YD":2,
    }
    colorMap = [    #颜色映射表
        [255  ,255    ,255],
        [47   ,196    ,72],
        [192  ,52     ,217],
    ]
    
    
    #featureLayer = arcpy.MakeFeatureLayer_management(inputPath, "featureLayer")
    
    #if 'DDL' in fields:
    # continue
    #else:
    # print("shp文件中不包含DDL字段")
    # sys.exit()
    
    #查询相应的字段并创建矢量文件
    arcpy.env.workspace = os.path.dirname(inputPath)
    try:
        print("开始添加字段！")
        arcpy.AddMessage("开始添加字段!")
        #查询相应的字段并创建矢量文件
        feature_set = arcpy.FeatureSet(inputPath)        #创建要素据集
        dirName = os.path.dirname(inputPath)
        fileName = os.path.basename(inputPath).split('.')[0]
        classDataOutputPath = os.path.join(outputPath,fileName+"_AddFields.shp")
        feature_set.save(classDataOutputPath)
        del feature_set
    
    	#向矢量文件中添加字段
        arcpy.AddField_management(classDataOutputPath,'Classname', 'TEXT', 50, '', '')
        arcpy.AddField_management(classDataOutputPath,'Classcode', 'TEXT',  50, '', '')
        arcpy.AddField_management(classDataOutputPath,'Classvalue', 'SHORT', None, 0, '')
        arcpy.AddField_management(classDataOutputPath,'RED', 'SHORT',  None, 0, '')
        arcpy.AddField_management(classDataOutputPath,'GREEN', 'SHORT',  None, 0, '')
        arcpy.AddField_management(classDataOutputPath,'BLUE', 'SHORT', None, 0, '')
    
        searchFields = [inputFields,"Classname","Classcode","Classvalue","RED","GREEN","BLUE"]
        with arcpy.da.UpdateCursor(classDataOutputPath,searchFields) as cursor:      #利用搜索游标来遍历相应字段并更新其内容
            for row in cursor:
                #print("{}".format(row))
                classes = classMap[row[0]]
                color = colorMap[classes]
                row[1] = row[0]
                row[2] = "clsss" + str(classes)
                row[3] = classes
                row[4] = color[0]
                row[5] = color[1]
                row[6] = color[2]
                #print("{}\n".format(row))
                cursor.updateRow(row)
    
        print("添加字段成功！")
        arcpy.AddMessage("添加字段成功!")
    except arcpy.ExecuteError:
        print(arcpy.GetMessages())

之后我们可以到Arcmap中进行配置一下（建立工具箱），就可以正常使用了，非常方便。

![在这里插入图片描述][2c8dfa3393f14e86afbd0bfa965e0097.png]

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBAZGF5dWhhaXRhbmdfZ2FsYXh5_size_15_color_FFFFFF_t_70_g_se_x_16]

实现的效果：

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBAZGF5dWhhaXRhbmdfZ2FsYXh5_size_11_color_FFFFFF_t_70_g_se_x_16]

[2c8dfa3393f14e86afbd0bfa965e0097.png]: /images/20220829/d551e9d334b14822b76fbb47c0c5b208.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBAZGF5dWhhaXRhbmdfZ2FsYXh5_size_15_color_FFFFFF_t_70_g_se_x_16]: /images/20220829/23f4269bb3b74f0fb537fe3176e27181.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBAZGF5dWhhaXRhbmdfZ2FsYXh5_size_11_color_FFFFFF_t_70_g_se_x_16]: /images/20220829/afe4f5ea89c04474b71f7bd49345aff2.png