我有一个包含数字字符串的列表,如下所示:
numbers = ['1', '5', '10', '8'];
我想将每个列表元素转换为整数,所以它看起来像这样:
numbers = [1, 5, 10, 8];
我可以使用循环来完成它,如下所示:
new_numbers = [];
for n in numbers:
new_numbers.append(int(n));
numbers = new_numbers;
它必须如此难看吗?我确信在一行代码中有更多的pythonic方法可以做到这一点。请帮帮我。
答案 0 :(得分:317)
这是list comprehensions的用途:
numbers = [ int(x) for x in numbers ]
答案 1 :(得分:119)
在Python 2.x中,另一种方法是使用map
:
numbers = map(int, numbers)
注意:在Python 3.x map
中返回一个地图对象,如果需要,可以将其转换为列表:
numbers = list(map(int, numbers))
答案 2 :(得分:17)
只是一点,
numbers = [int(x) for x in numbers]
列表理解更自然,而
numbers = map(int, numbers)
更快。
在大多数情况下,这可能无关紧要
有用的读物:LP vs map
答案 3 :(得分:9)
如果您打算将这些整数传递给函数或方法,请考虑以下示例:
sum(int(x) for x in numbers)
这种结构有意与adamk提到的列表理解非常相似。没有方括号,它被称为生成器表达式,并且是一种将参数列表传递给方法的一种非常节省内存的方法。这里有一个很好的讨论:Generator Expressions vs. List Comprehension
答案 4 :(得分:3)
另一种方式,
for i, v in enumerate(numbers): numbers[i] = int(v)
答案 5 :(得分:3)
在Python 3中实现它的另一种方法:
numbers = [*map(int, numbers)]
答案 6 :(得分:0)
我认为我会整合答案并显示一些timeit
个结果。
Python 2很糟糕,但map
比理解快一点。
Python 2.7.13 (v2.7.13:a06454b1afa1, Dec 17 2016, 20:42:59) [MSC v.1500 32 bit (Intel)] on win32
Type "copyright", "credits" or "license()" for more information.
>>> import timeit
>>> setup = """import random
random.seed(10)
l = [str(random.randint(0, 99)) for i in range(100)]"""
>>> timeit.timeit('[int(v) for v in l]', setup)
116.25092001434314
>>> timeit.timeit('map(int, l)', setup)
106.66044823117454
Python 3本身速度提高了4倍,但将map
生成器对象转换为列表仍然比理解更快,并通过解压缩map
生成器创建列表(感谢Artem!)是还是稍快一点。
Python 3.6.1 (v3.6.1:69c0db5, Mar 21 2017, 17:54:52) [MSC v.1900 32 bit (Intel)] on win32
Type "copyright", "credits" or "license()" for more information.
>>> import timeit
>>> setup = """import random
random.seed(10)
l = [str(random.randint(0, 99)) for i in range(100)]"""
>>> timeit.timeit('[int(v) for v in l]', setup)
25.133059591551955
>>> timeit.timeit('list(map(int, l))', setup)
19.705547827217515
>>> timeit.timeit('[*map(int, l)]', setup)
19.45838406513076
注意:在Python 3中,4个元素似乎是交叉点(Python 2中的3个),其中理解稍快一些,尽管解包生成器仍然比具有多于1个元素的列表更快。