我有一个包含数字字符串的列表,如下所示:
numbers = ['1', '5', '10', '8'];
我想把每个列表元素都转换成整数,所以它看起来像这样:
numbers = [1, 5, 10, 8];
我可以用循环来做,像这样:
new_numbers = [];
for n in numbers:
new_numbers.append(int(n));
numbers = new_numbers;
非得这么难看吗?我相信有一种更python化的方法可以在一行代码中实现这一点。请帮帮我。
我想我应该巩固答案,并展示一些时间的结果。
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倍以上,但是将映射生成器对象转换为列表仍然比理解快,并且通过解包映射生成器创建列表(感谢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个元素的列表,解包生成器仍然比任何一个都快。