Create a Python List with Multiple Elements per Iteration

Question

def f1(x): return [(x+1)*2-1, (x+1)*2-1]
def f2(x): return [(x+1)*2, (x+1)*2]

[[f1(i), f2(i)] for i in np.arange(3)]

This is the code to generate a list of 3 list-pairs elements:

[[[1, 1], [2, 2]], [[3, 3], [4, 4]], [[5, 5], [6, 6]]]

However, I would like to obtain a list like below with one line of for loop.

[[1, 1], [2, 2], [3, 3], [4, 4], [5, 5], [6, 6]]

This is how it works with multi-lines:

n = []
for i in np.arange(3):
    n += [f1(i), f2(i)]

It's like trying to compose 2 elements per time where I don't know how to achieve += for one line of code. How can I do that?

user2357112 · Accepted Answer · 2016-05-05 00:29:47Z

2

[x for i in np.arange(3) for x in [f1(i), f2(i)]]

Use a list comprehension with two for clauses.

answered May 5, 2016 at 0:29

user2357112

286k32 gold badges490 silver badges569 bronze badges

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

2 Comments

João Oliveira Over a year ago

Thank you! I think this is far one-line of code can go. But [x for x in [f1(i), f2(i)] for i in np.arange(3)] also works. That means the for order is optional?

user2357112 Over a year ago

The loop order matters. What you wrote does something entirely different, and probably not what you want.

João Oliveira · Accepted Answer · 2016-05-05 00:23:51Z

0

I could do something like this:

[f1(i) for i in np.arange(3)] + [f2(i) for i in np.arange(3)]

But is there any better way?

answered May 5, 2016 at 0:23

João Oliveira

4523 silver badges18 bronze badges

Collectives™ on Stack Overflow

Create a Python List with Multiple Elements per Iteration

2 Answers 2

2 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related