Loop in Python List using more than one item per loop

Question

So I'm writing a script for a program called Abaqus.... and I have a list of numbers, and I need to loop through the numbers in the following manner

listOfSteps = [1, 4, 7, 10, 17, 22, 28, 29, 30, 43, 47, 50]
fileNameCreate = 0 

for i in listOfSteps:

    session.viewports['Viewport: 1'].odbDisplay.setFrame(step=i, frame=-1)
    session.viewports['Viewport: 2'].odbDisplay.setFrame(step=i, frame=-1)
    session.viewports['Viewport: 3'].odbDisplay.setFrame(step=i, frame=-1)
    session.printOptions.setValues(reduceColors=False)
    session.printToFile(fileName='C:/Image'+str(fileNameCreate+1), format=PNG, 
        canvasObjects=(session.viewports['Viewport: 3'], 
        session.viewports['Viewport: 2'], session.viewports['Viewport: 1']))

So I need the first step to use 1, 2nd step to use 4, 3rd step to use 7
Then do the code to save the file

Then start the loop again at 10

Any help would be great.

Aaron Cronin · Accepted Answer · 2015-02-17 13:59:58Z

5

Assuming I've understood your question correctly, you could use an iterator:

listOfSteps = [1, 4, 7, 10, 17, 22, 28, 29, 30, 43, 47, 50]
fileNameCreate = 0 

it = iter(listOfSteps)

for a in it:
    b = next(it)
    c = next(it)

    session.viewports['Viewport: 1'].odbDisplay.setFrame(step=a, frame=-1)
    session.viewports['Viewport: 2'].odbDisplay.setFrame(step=b, frame=-1)
    session.viewports['Viewport: 3'].odbDisplay.setFrame(step=c, frame=-1)

    # ...

And, if you don't mind a little bit of magic:

for a, b, c in zip(*[iter(listOfSteps)]*3):
    # ...

The left-to-right evaluation order of the iterables is guaranteed. This makes possible an idiom for clustering a data series into n-length groups using zip(*[iter(s)]*n). ~ Zip Docs

edited Feb 17, 2015 at 13:59

answered Feb 17, 2015 at 13:54

Aaron Cronin

2,12314 silver badges13 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

frlan · Accepted Answer · 2015-02-17 13:55:07Z

0

Maybe try to add a counter variable to your code

listOfSteps = [1, 4, 7, 10, 17, 22, 28, 29, 30, 43, 47, 50]
fileNameCreate = 0 
cnt = 0

for i in listOfSteps:
   cnt =+ 1
   if cnt % 3 == 0:
       # here do your write magic

answered Feb 17, 2015 at 13:55

frlan

7,2953 gold badges36 silver badges72 bronze badges

1 Comment

user4237459 Over a year ago

for cnt, i in enumerate(listOfSteps): would achieve the same thing without manually counting up.

Collectives™ on Stack Overflow

Loop in Python List using more than one item per loop

2 Answers 2

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related