Python嵌套更新游标与嵌套字典。必须有一个更简单的方法

时间:2016-01-02 06:59:40

标签: python dictionary cursor arcmap

我创建了一个从表中调用值的嵌套字典,我需要使用该数据更新要素类的属性表。我让它使用两个硬编码字段作为测试,但我需要弄清楚如何自动获取featFields的长度并使用它来指示要更新的每个字段的索引位置。因此,我不是硬编码row[1]row[2]等,而是'LOCDESC''RIMELEV',而是使用变量来逐步调整每个变量的索引位置。

我在Python工作。最终目标是在ArcMap 10.2或10.3中使用的工具箱。

import arcpy
arcpy.env.workspace = r"C:/SARP10/MACP_Tool"

#Define fields to update and the field to use as join field

Table = "Test2.csv"
Input = "Test.gdb/MHs"
csvFields = ['Location_Details', 'Elevation']
featFields = ['LOCDESC', 'RIMELEV']
csvKey = "Manhole_Number"
featKey = "FACILITYID"
csvFields.insert(0, csvKey)
featFields.insert(0, featKey)
print csvFields
#Create dictionary to store values from the update table
UpdateDict = {}

#Iterates through the values in the table and stores them in UpdateDict

with arcpy.da.SearchCursor(Table, csvFields) as cursor:
    for row in cursor:
        UpdateDict[row[0]] = dict(zip(featFields[1:], row[1:]))
    print UpdateDict

MHNum = len(UpdateDict) # gets # of MHs to be updated
MHKeys = UpdateDict.keys() # gets key values, i.e. MH numbers

print "You are updating fields for the following {} manholes: {}".format(MHNum, MHKeys)

#Iterates through feature class attribute table and updates desired attributes
with arcpy.da.UpdateCursor(Input, featFields) as cursor:
    i = 0
    z = 0

    for row in cursor:
        i += 1
        for f in UpdateDict.keys():
            if f == row[0]:
                row[1] = UpdateDict.values()[z]['LOCDESC']#uses counter and subdict key to call correct value
                row[2] = UpdateDict.values()[z]['RIMELEV']#uses counter and subdict key to call correct value
                cursor.updateRow(row)
                z +=1 #counter keeps track of rows and provides index location for dictionary
                print "Updating {} of {} manholes in this submittal: {}.".format(z, MHNum, f)
            else:
                pass
print "Updated {} of {} rows.".format(MHNum, i)
print "Script completed."

2 个答案:

答案 0 :(得分:0)

由于row[n]的{​​当前硬编码'迭代逐步遍历featFields的值,您可以设置一个for循环,迭代它们两者,如:

if f == row[0]:
    # loop set by length of featFields list
    for j in range(0, len(featFields) - 1):
        row[j + 1] = UpdateDict.values()[z][featFields[j]]
        cursor.updateRow(row)
        # etc.

请注意“偏移量” - row[1]应该使用featFields[0]等等 - 需要考虑。

答案 1 :(得分:0)

问题 ,访问数据字典中的正确字段。最终代码访问外键列表和内键:值对的列表,其中变量(z)设置为保持两个列表中的索引号相等。谢谢你的帮助,@ Erica!

以下是有效的:

import arcpy
arcpy.env.workspace = r"C:/SARP10/MACP_Tool"

#Defines fields to update and the field to use as join field
Table = "Test2.csv"
Input = "Test.gdb/MHs"
csvFields = ['Location_Details', 'Elevation', 'Rim_to_Invert', 'Rim_to_Grade', 'Cover_Size', 'Wall_Material', 'Wall_Diam', 'Wall_Lining_Interior', 'Photo2_Link', 'MH_InspectReportLink'] #update table fields
featFields = ['LOCDESC', 'RIMELEV', 'RIMTOINVERT', 'RIMTOGRADE','COVERSIZE','WALLMAT','DIAMETER','LINERTYPE','HYPERLINK_PHOTO2','HYPERLINK_RPT']#fc field names
csvKey = "Manhole_Number"
featKey = "FACILITYID"
csvFields.insert(0, csvKey)
featFields.insert(0, featKey)
print "Your table contains the following fields to be updated: {}\n".format(str(csvFields))

#Process: Create dictionary to store values from the update table, iterate through values and store in UpdateDict
UpdateDict = {}

with arcpy.da.SearchCursor(Table, csvFields) as cursor:
   for row in cursor:
      UpdateDict[row[0]] = dict(zip(featFields[1:], row[1:]))
## debug print "You have created update dictionary 'UpdateDict': \n{}\n\n".format(UpdateDict)

MHNum = len(UpdateDict) # gets # of MHs to be updatedMHKeys = sorted(UpdateDict.keys()) # gets key values, i.e. MH numbers
MHKeys = UpdateDict.keys() #calls outer keys (MH numbers, which are join values) into a list of keys
MHVals = UpdateDict.values()#calls inner nested key:value pairs to a list

##debug print "Dictionary keys: {}\n\n Dictionary values: {}\n\n".format(str(MHKeys),str(MHVals))
print "You are updating fields for the following {} manholes: {}".format(MHNum, str(MHKeys))

#Process: Iterates through feature class attribute table and updates desired attributes
with arcpy.da.UpdateCursor(Input, featFields) as curs:
    i = 0 #attribute table row counter
    for row in curs:
        i += 1
        for f in MHKeys:
            if f == row[0]:
                z = MHKeys.index(f)#get index location in MHKeys
                for y in range(0,len(featFields)-1):
                    row[y+1] = MHVals[z][featFields[y+1]]#use z to pull corresponding value in MHVals to correct key in MHKeys
                print "Current MH: {} \nUpdating Values: {} \n\n".format(f, UpdateDict.values()[z])
                curs.updateRow(row)
            else:
                pass

print "Updated {} of {} rows.".format(MHNum, i)
print "Script completed."