【Table攻略】多键-值对最值筛选代码

#代码背景#建立一个6*6*6的立方体，提取各个z坐标下的单元id，找出各个z坐标下的最大单元id并单独输出为excelimport itasca as itimport pandas as pdit.command("python-reset-state false")#建立模型it.command("""           model new           zone create brick size 6 6 6           zone densify           """)#创建一个空字典zid和一个空列表zposzid = {}zpos = []#遍历单元，将ID-z坐标以键-值对的方式存放在zid中，将各单元的z坐标存放在zpos中for z in it.zone.list():    zid[z.id()]=z.pos_z()    zpos.append(z.pos_z())#对z坐标去重new_zpos = list(set(zpos))#获取z坐标个数j = len(new_zpos)#创建一个空字典new_zidnew_zid = {}#遍历z坐标，将同一z坐标对应的单元id存放于列表keys中并获取最大值，然后将z坐标-最大ID以键-值对的方式存放于字典new_zid中for i in range(j):    keys = []    for key in zid.keys():        if zid[key] == new_zpos[i]:            keys.append(key)    key_max = max(keys)    new_zid[new_zpos[i]] = key_max#将字典new_zid中的键值对以z坐标从小到大进行排序sorted_new_zid = dict(sorted(new_zid.items(),key=lambda x:x[0]))#将字典new_zid中单元ID、z坐标分别存放在列表id_list、pos_list中id_list = list(sorted_new_zid.values())pos_list = list(sorted_new_zid.keys())#创建空字典toExcel，并将上述列表id_list、pos_list存放在空字典toExcel中toExcel = {}toExcel['ZPOS'] = pos_listtoExcel['ID'] = id_list#通过pandas库输出字典toExceldf = pd.DataFrame(toExcel)df.to_excel('ceshi.xlsx',index=False)