Python改变整个列表的类型？

Question

我想做这样的事情

def foo(x,dtype=long):
   return magic_function_changing_listtype_to_dtype(x)

即一个完整的列表到一个完整的int列表

任何简单的方法来嵌套列表,即更改类型[['1'],['2']] - > int

Answer 1

Python 2:

map(int, ['1','2','3']) # => [1,2,3]

...

def foo(l, dtype=long):
    return map(dtype, l)

在Python 3中,map()返回一个map对象,因此你需要将它转换为一个列表:

list(map(int, ['1','2','3'])) # => [1,2,3]

...

def foo(l, dtype=long):
    return list(map(dtype, l))

Answer 2

这是一个非常简单的递归函数,用于转换任何深度的嵌套列表:

def nested_change(item, func):
    if isinstance(item, list):
        return [nested_change(x, func) for x in item]
    return func(item)

>>> nested_change([['1'], ['2']], int)
[[1], [2]]
>>> nested_change([['1'], ['2', ['3', '4']]], int)
[[1], [2, [3, 4]]]

Answer 3

列表理解应该这样做:

a = ['1','2','3']
print [int(s) for s in a]   # [1, 2, 3]]

嵌套:

a = [['1', '2'],['3','4','5']]
print [[int(s) for s in sublist] for sublist in a]   # [[1, 2], [3, 4, 5]]