13

In Python, I am trying to apply an operator to a two layer nested array. For example, 

a = [['2.3','.2'],['-6.3','0.9']]
for j in range(2)
    for i in range(2)
        a[i][j] = float(a[i][j])

How can I do this without the loops? I am hoping for something akin to a= map(float,a). Of course the last script does not work for a nested list. A one line list comprehension may be acceptable too.

Hans
  • 1,269
  • 3
  • 19
  • 38

4 Answers4

14

One-liner with mix of map and listcomp:

a = [map(float, suba) for suba in a]  # Only works on Py2

Or variants:

# Both of the below work on Py2 and Py3
a = [list(map(float, suba)) for suba in a]
a = [[float(x) for x in suba] for suba in a]

Choose based on your personal preference and target Python version. For large nested lists on CPython 2, the first variant is probably the fastest (if the inner lists are huge, it avoids lookup overhead to get the float constructor and byte code execution for the inner lists), and the list wrapped equivalent might eventually win on CPython 3; for small nested lists on all versions, the nested list comprehensions is usually going to be the fastest.

ShadowRanger
  • 143,180
  • 12
  • 188
  • 271
8

With list comprehension:

a =  [[float(j) for j in i]  for i in a]
Iron Fist
  • 10,739
  • 2
  • 18
  • 34
4

The first argument requires a function, and as such, you can utilize the lambda operator to map float to the sub-lists. This is an example:

a = map(lambda b : map(float, b), a)
szabad
  • 380
  • 3
  • 10
  • 1
    FYI, the sole advantage of `map` over list comprehensions (brevity and speed when applying existing built-in functions) is lost when you use it with `lambda` functions (because each call to the `lambda` function adds overhead that inlining the same work in a list comprehension avoids completely). – ShadowRanger Dec 04 '15 at 04:27
2

Under the hood, map is a loop too. You should just use a list comprehension, however there is a nice way to do it with Python2 - you just need a partial function

>>> a = [['2.3','.2'],['-6.3','0.9']]
>>> from functools import partial
>>> map(partial(map, float), a)
[[2.3, 0.2], [-6.3, 0.9]]

I don't think there is a nice way to convert this to Python3 though

>>> list(map(list, map(partial(map, float), a)))  #YUKKK!
[[2.3, 0.2], [-6.3, 0.9]]
John La Rooy
  • 295,403
  • 53
  • 369
  • 502