在Python 3中连接序列的首选方法是什么?
现在,我正在做:
import functools
import operator
def concatenate(sequences):
return functools.reduce(operator.add, sequences)
print(concatenate([['spam', 'eggs'], ['ham']]))
# ['spam', 'eggs', 'ham']
需要导入两个单独的模块才能做到这一点似乎很笨拙。
另一种选择可能是:
def concatenate(sequences):
concatenated_sequence = []
for sequence in sequences:
concatenated_sequence += sequence
return concatenated_sequence
但是,这是不正确的,因为您不知道序列是列表。
你可以这样做:
import copy
def concatenate(sequences):
head, *tail = sequences
concatenated_sequence = copy.copy(head)
for sequence in sequences:
concatenated_sequence += sequence
return concatenated_sequence
但这似乎很容易出错 - 直接调用复制? (我知道head.copy()
适用于列表和元组,但是copy
不是序列ABC的一部分,所以你不能依赖它......如果你得到了字符串怎么办?)。如果您获得MutableSequence
,您必须复制以防止变异。此外,此解决方案会强制您首先解压缩整个序列集。再试一次:
import copy
def concatenate(sequences):
iterable = iter(sequences)
head = next(iterable)
concatenated_sequence = copy.copy(head)
for sequence in iterable:
concatenated_sequence += sequence
return concatenated_sequence
但是来吧......这是蟒蛇!那么......这样做的首选方法是什么?
答案 0 :(得分:8)
我改用itertools.chain.from_iterable()
:
import itertools
def chained(sequences):
return itertools.chain.from_iterable(sequences):
或者,由于您使用python-3.3对其进行了标记,因此可以使用新的yield from
语法(看看ma,没有导入!):
def chained(sequences):
for seq in sequences:
yield from seq
它们都返回迭代器(如果必须实现完整列表,则使用list()
)。大多数情况下,您不需要一系列连接序列,实际上,您只需要循环它们来处理和/或搜索某些内容。
请注意,对于字符串,您应该使用str.join()
而不是我的答案或您的问题中描述的任何技术:
concatenated = ''.join(sequence_of_strings)
结合起来,为了处理 fast 的序列并纠正,我会使用:
def chained(sequences):
for seq in sequences:
yield from seq
def concatenate(sequences):
sequences = iter(sequences)
first = next(sequences)
if hasattr(first, 'join'):
return first + ''.join(sequences)
return first + type(first)(chained(sequences))
这适用于元组,列表和字符串:
>>> concatenate(['abcd', 'efgh', 'ijkl'])
'abcdefghijkl'
>>> concatenate([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
[1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> concatenate([(1, 2, 3), (4, 5, 6), (7, 8, 9)])
(1, 2, 3, 4, 5, 6, 7, 8, 9)
并对字符串序列使用更快的''.join()
。
答案 1 :(得分:1)
使用itertools.chain.from_iterable
。
import itertools
def concatenate(sequences):
return list(itertools.chain.from_iterable(sequences))
只有当您需要一个实际的新列表时才需要调用list
,所以如果您只是迭代这个新序列一次就跳过它。
答案 2 :(得分:0)
怎么了?
from itertools import chain
def chain_sequences(*sequences):
return chain(*sequences)