Python multiple for loops

Question

I am wondering if the 3 for loops in the following code can be written in a better way:

   Nc = 10     # number of points for (0, pi)
   cc1 = linspace(0,pi,Nc)
   cc2 = linspace(0,pi/2,Nc/2)
   cc3 = linspace(0,pi/2,Nc/2)
   for c1 in cc1:
       for c2 in cc2:
           for c3 in cc3:
               print c1,c2,c3

it is not a duplicate. The answer based on numpy.meshgrid() or similar is not appropriate for the other question. — jfs
– jfs, Commented Aug 20, 2012 at 4:21
Better in what way? Faster? More memory efficient? Prettier? — bchurchill
– bchurchill, Commented Aug 20, 2012 at 5:48

Amber · Accepted Answer · 2012-08-20 04:13:00Z

28

import itertools

for a,b,c in itertools.product(cc1, cc2, cc3):
    print a,b,c

answered Aug 20, 2012 at 4:13

Amber

531k89 gold badges643 silver badges558 bronze badges

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

2 Comments

Camilo Over a year ago

Should be : print( (a,b,c)

Glatinis Over a year ago

This is Python 2.

blueiur · Accepted Answer · 2012-08-20 04:13:06Z

8

try this :)

[(c1, c2, c3) for c1 in cc1 for c2 in cc2 for c3 in cc3]

answered Aug 20, 2012 at 4:13

blueiur

1,5071 gold badge11 silver badges17 bronze badges

Collectives™ on Stack Overflow

Python multiple for loops

2 Answers 2

2 Comments

Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Linked

Related