Python3 中的map

Posted 2020-12-11 那抹阳光1994

最近看一段Python2写的代码，有一句：

torch.FloatTensor(map(lambda x: x[‘values‘], data))

Python3下运行后报错：

TypeError: new(): data must be a sequence (got map)

我的data明明是sequence！原来是Python版本的问题ε=(´ο｀*)))

 

map() 会根据提供的函数对指定序列做映射。

语法：
map() 函数语法：map(function, iterable, …)

参数：
function – 函数
iterable – 一个或多个序列

第一个参数 function 以参数序列中的每一个元素调用 function 函数，返回包含每次 function 函数返回值的新列表。

返回值：

Python 2.x 返回列表。
Python 3.x 返回迭代器。

例如：我想返回由字典组成的list a中的key为key1的所有元素的值，直接采用map返回一个iterator，通过list(map())将map转换为list。

 1 a = [{‘key1‘:1, ‘key2‘:2}, {‘key1‘: 3, ‘key3‘: 4}]
 2 
 3 b = map(lambda x: x[‘key1‘],  a)
 4 
 5 >>> b
 6 
 7 <map at 0x23f3e386188>
 8 
 9 
10 >>> b = list(map(lambda x: x[‘key1‘], a))
11 
12 [1, 3]