list comprehension question

Chris Rebert clp2 at rebertia.com
Thu Apr 30 21:05:42 EDT 2009


On Thu, Apr 30, 2009 at 5:56 PM, Ross <ross.jett at gmail.com> wrote:
> If I have a list of tuples a = [(1,2), (3,4), (5,6)], and I want to
> return a new list of each individual element in these tuples, I can do
> it with a nested for loop but when I try to do it using the list
> comprehension b = [j for j in i for i in a], my output is b =
> [5,5,5,6,6,6] instead of the correct b = [1,2,3,4,5,6]. What am I
> doing wrong?

Your comprehension is the identity comprehension (i.e. it effectively
just copies the list as-is).
What you're trying to do is difficult if not impossible to do as a
comprehension.

Here's another approach:
b = list(itertools.chain.from_iterable(a))

And without using a library function:
b = []
for pair in a:
    for item in pair:
        b.append(item)

Cheers,
Chris
-- 
http://blog.rebertia.com



More information about the Python-list mailing list