Итерация по трем спискам одновременно в Python? - PullRequest
Новая
       10

Итерация по трем спискам одновременно в Python?

6 голосов
/ 27 июня 2011

Это может быть довольно сложный вопрос, поскольку многие из вас могут не знать, для какого программного обеспечения я пишу это: Autodesk Maya 2011. Я пытаюсь ускорить утомительный медленный процесс символы способности), написав скрипт, который делает это автоматически.

Я постараюсь объяснить ситуацию.

У меня есть сценарий, который берет объект, перебирает дочерние объекты этого объекта, сохраняет их в списке, затем помещает начальный объект в конец списка, переворачивает список, потому что это неправильно, и помещает начальный объект спереди.

Проблема: существует три разных списка одного и того же ТИПА объекта, но с разными именами, и на самом деле это разные объекты. Моя цель - соединить их вместе, генерируя узлы, называемые «blendcolors». Но если у меня есть цикл для генерации их для каждого объекта в списке A, то мне нужны циклы, которые также связывают их с объектами в других списках, и я не могу понять это.

Вот мой код, с которым он был воспроизведен, поэтому он более неполный, чем раньше, в том, что касается самого цикла. 

    import maya.cmds as cmds

    def crBC(IKJoint, FKJoint, bindJoint, xQuan, switch):

        # gets children joints of the selected joint
        chHipIK = cmds.listRelatives(IKJoint, ad = True, type = 'joint')
        chHipFK = cmds.listRelatives(FKJoint, ad = True, type = 'joint')
        chHipBind = cmds.listRelatives(bindJoint, ad = True, type = 'joint')
        # list is built backwards, this reverses the list
        chHipIK.reverse()
        chHipFK.reverse()
        chHipBind.reverse()
        # appends the initial joint to the list
        chHipIK.append(IKJoint)
        chHipFK.append(FKJoint)
        chHipBind.append(bindJoint)
        # puts the last joint at the start of the list because the initial joint
        # was added to the end
        chHipIK.insert(0, chHipIK.pop())
        chHipFK.insert(0, chHipFK.pop())
        chHipBind.insert(0, chHipBind.pop())


        # pops off the remaining joints in the list the user does not wish to be blended
        chHipBind[xQuan:] = []

        chHipIK[xQuan:] = []

        chHipFK[xQuan:] = []

       # goes through the bind joints, makes a blend colors for each one, connects
       # the switch to the blender

        for a in chHipBind


            rotBC = cmds.shadingNode('blendColors', asUtility = True, n = a + 'rotate_BC')
            tranBC = cmds.shadingNode('blendColors', asUtility = True, n = a + 'tran_BC')
            scaleBC = cmds.shadingNode('blendColors', asUtility = True, n = a + 'scale_BC')

            cmds.connectAttr(switch + '.ikFkSwitch', rotBC + '.blender')
            cmds.connectAttr(switch + '.ikFkSwitch', tranBC + '.blender')
            cmds.connectAttr(switch + '.ikFkSwitch', scaleBC + '.blender')

        # goes through the ik joints, connects to the blend colors

            for b in chHipIK:
                cmds.connectAttr(b + '.rotate', rotBC + '.color1')
                cmds.connectAttr(b + '.translate', tranBC + '.color1')
                cmds.connectAttr(b + '.scale', scaleBC + '.color1')


            # connects FK joints to the blend colors

            for c in chHipFK:
                cmds.connectAttr(c + '.rotate', rotBC + '.color2')
                cmds.connectAttr(c + '.translate', tranBC + '.color2')
                cmds.connectAttr(c + '.scale', scaleBC + '.color2')

        # connects blend colors to bind joints


            cmds.connectAttr(rotBC + '.output', d + '.rotate')
            cmds.connectAttr(tranBC + '.output', d + '.translate')
            cmds.connectAttr(scaleBC + '.output', d + '.scale')                






    # executes function


    crBC('L_hip_IK', 'L_hip_FK', 'L_hip_JNT', 6, 'L_legSwitch_CTRL')

Ответы [ 3 ]

27 голосов
/ 27 июня 2011

Я не совсем понимаю вопрос, вы ищете 

import itertools
for a, b, c in itertools.izip(lst1, lst2, lst3):
    ...

?

Что делает izip, это принимает переменное число аргументов и возвращает итератор, который всегдавозвращает соответствующие элементы аргументов (кортеж первых аргументов в первом прогоне, кортеж вторых аргументов во втором прогоне и т. д. и т. д.).

1 голос
/ 13 сентября 2016

@ florian mayer way с zip или izip работает отлично, но вы также можете использовать перечисление, чтобы получить счетчик списка, 

list_a  = ["x", "y"]
list_b  = ["k", "j"]
list_c  = ["m", "n"]

for count, item in enumerate(list_a):
    print item, list_b[count], list_c[count]
1 голос
/ 27 июня 2011

Почему существуют "Есть три разных списка одного и того же типа объекта"? Почему это нельзя исправить, чтобы создать один список, в котором все три объекта сопоставлены правильно?

Скорее всего, простое отображение будет лучше, чем три параллельных списка.

В частности, вам нужно исправить chHipIK = cmds.listRelatives(IKJoint, ad = True, type = 'joint') для такой работы. 

chHipIK = [ { 'IK': ik } for ik in mds.listRelatives(IKJoint, ad = True, type = 'joint') ]
for i, fk in enumerate(cmds.listRelatives(FKJoint, ad = True, type = 'joint')):
    chHipIK[i]['FK']= fk
for i, bind in enumerate(cmds.listRelatives(bindJoint, ad = True, type = 'joint')):
    chHipIK[i]['FK']= bind

Так что chHipIK - это список отображений, в котором есть все три элемента информации.

...