使用您的列表组合条is调用两次,如果您只想调用 strip 一次并保持理解,请使用 gen exp:
input_list[:] = [x for x in (s.strip() for s in input_list) if x]
Input:
input_list = [' \n ',' data1\n ',' data2\n',' \n','data3\n']
Output:
['data1', 'data2', 'data3']
input_list[:]
将更改原始列表,这可能是也可能不是您想要的,如果您实际上想创建一个新列表,只需使用cleaned_up_list = ...
.
我总是发现使用itertools.imap
在Python 2和map
在 python 3 中,对于较大的输入,而不是生成器是最有效的:
from itertools import imap
input_list[:] = [x for x in imap(str.strip, input_list) if x]
一些采用不同方法的计时:
In [17]: input_list = [choice(input_list) for _ in range(1000000)]
In [19]: timeit filter(None, imap(str.strip, input_list))
10 loops, best of 3: 115 ms per loop
In [20]: timeit list(ifilter(None,imap(str.strip,input_list)))
10 loops, best of 3: 110 ms per loop
In [21]: timeit [x for x in imap(str.strip,input_list) if x]
10 loops, best of 3: 125 ms per loop
In [22]: timeit [x for x in (s.strip() for s in input_list) if x]
10 loops, best of 3: 145 ms per loop
In [23]: timeit [data.strip() for data in input_list if data.strip()]
10 loops, best of 3: 160 ms per loop
In [24]: %%timeit
....: cleaned_up_list = []
....: for data in input_list:
....: clean_data = data.strip()
....: if clean_data:
....: cleaned_up_list.append(clean_data)
....:
10 loops, best of 3: 150 ms per loop
In [25]:
In [25]: %%timeit
....: cleaned_up_list = []
....: append = cleaned_up_list.append
....: for data in input_list:
....: clean_data = data.strip()
....: if clean_data:
....: append(clean_data)
....:
10 loops, best of 3: 123 ms per loop
最快的方法实际上是itertools.ifilter
结合itertools.imap
紧随其后的是filter
with imap
.
无需重新评估函数引用list.append
每次迭代都更有效,如果您陷入循环并想要最有效的方法,那么它是一个可行的替代方案。