Python3 中的map

最近看一段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。

a = [{'key1':1, 'key2':2}, {'key1': 3, 'key3': 4}]

b = map(lambda x: x['key1'],  a)

>>> b

<map at 0x23f3e386188>


>>> b = list(map(lambda x: x['key1'], a))

[1, 3]