如何将函数映射到列表的每个子列表中的每个项目

Question

有没有一种方法可以在不使用常规for循环遍历主列表的情况下执行此操作？

>>> map(lambda x: x*2, [[1,2,3],[4,5,6]]) 
[[1, 2, 3, 1, 2, 3], [4, 5, 6, 4, 5, 6]] 
# want [[2,4,6],[8,10,12]] 

Answer 1

您有嵌套列表，x只是其中一个列表。为了处理这一点，你需要真正的上的x个别元素映射乘法功能，这样

>>> map(lambda x: map(lambda y: y * 2, x), [[1, 2, 3], [4, 5, 6]]) 
[[2, 4, 6], [8, 10, 12]] 

但我宁愿在这个列表理解，

>>> [[y * 2 for y in x] for x in [[1, 2, 3], [4, 5, 6]]] 
[[2, 4, 6], [8, 10, 12]] 

Answer 2

替代的解决方案是去块状矢量化操作：

import numpy as np 

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

(2*np.array(ll)).tolist() 
#Out[6]: [[2, 4, 6], [8, 10, 12]] 

Answer 3

这是有点矫枉过正，不太实际这个特殊的例子，但另一种风格选项可以使用functools.partial来说明发生了什么以及map和列表理解的组合。

from functools import partial 
from operator import mul 

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

double = partial(mul, 2) 
dub_l = [map(double, sub) for sub in l]