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]

相关推荐