List comprehension on a nested list?

Boy Pasmo picture Boy Pasmo · Aug 6, 2013 · Viewed 159.5k times · Source

I have this nested list:

l = [['40', '20', '10', '30'], ['20', '20', '20', '20', '20', '30', '20'], ['30', '20', '30', '50', '10', '30', '20', '20', '20'], ['100', '100'], ['100', '100', '100', '100', '100'], ['100', '100', '100', '100']]

Now, what I want to do is convert each element in a list to float. My solution is this:

newList = []
for x in l:
  for y in x:
    newList.append(float(y))

But can this be done using nested list comprehension, right?

what I've done is:

[float(y) for y in x for x in l]

But then the result is bunch of 100's with the sum of 2400.

any solution, an explanation would be much appreciated. Thanks!

Answer

Andrew Clark picture Andrew Clark · Aug 6, 2013

Here is how you would do this with a nested list comprehension:

[[float(y) for y in x] for x in l]

This would give you a list of lists, similar to what you started with except with floats instead of strings. If you want one flat list then you would use [float(y) for x in l for y in x].