如何在 Python 中以特定方式重新组织列表

Question

所以，如果你有以下列表，我想做的是：

example_list=['This', 'is', 'QQQQQ', 'an', 'QQQQQ', 'example', 'list', 'QQQQQ', '.']

我希望它被重新组织成这样：

example_list=['This is', 'an', 'example list', '.']

注意 QQQQQ 是如何被用作占位符的。所以，基本上我希望 QQQQQ 之间的所有内容都成为一个列表元素。我该怎么做？

我看过其他关于 join() 函数的帖子，但我遇到的问题是如果超过 1 个单词，则在两者之间放置一个空格。

Answer 1

使用简单的迭代。

例如：

example_list=['This', 'is', 'QQQQQ', 'an', 'QQQQQ', 'example', 'list', 'QQQQQ', '.']

res = [[]]
for i in example_list:
    if i == "QQQQQ":
        res.append([])
    else:
        res[-1].append(i)
print([" ".join(i) for i in res])

输出：

['This is', 'an', 'example list', '.']

Answer 2

你可以使用itertools.groupby():

>>> from itertools import groupby
>>> example_list=['This', 'is', 'QQQQQ', 'an', 'QQQQQ', 'example', 'list', 'QQQQQ', '.']
>>> [' '.join(g) for k, g in groupby(example_list, lambda x: x == 'QQQQQ') if not k]
['This is', 'an', 'example list', '.']

或者甚至与 .__eq__ 比较，正如 cmets 中的 @tobias_k 所建议的那样：

>>> [' '.join(g) for k, g in groupby(example_list, key='QQQQQ'.__eq__) if not k]
['This is', 'an', 'example list', '.']

Answer 3

尝试将join 与strip() 一起消除空格

answer = [s.strip() for s in ' '.join(map(str, example_list)).split('QQQQQ')]
print (answer)

输出

['This is', 'an', 'example list', '.']

Answer 4

简单的解决方案：使用空格进行连接，然后将空格添加到拆分函数中的占位符。

示例：

example_list = ['This', 'is', 'QQQQQ', 'an', 'QQQQQ', 'example', 'list', 'QQQQQ', '.']

print(' '.join(example_list).split(' QQQQQ '))

结果：

['This is', 'an', 'example list', '.']

或更广义的：

split_arg = ' {} '.format(place_holder)
example_list = ' '.join(example_list).split(split_arg)

tobias_k 评论后编辑

评论是：“当然，这仅在占位符实际上是一个字符串时才有效，并且如果该搅拌没有出现在任何其他单词中。也就是说，如果占位符是，例如，None，则它将不起作用，' Q'，或''——tobias_k"

这是真的，所以我提出了一个更通用的解决方案，因此它应该适用于每个占位符。

import random
import string

example_list = ['This', 'is', None, 'an', None, 'example', 'list', None, '.']
place_holder = None
# create a random string of length 10
random_place_holder = ''.join(random.choices(string.ascii_uppercase + string.digits, k=10))  
# Replace all old place holders with our new random string placeholder
example_list = [x if x != place_holder else random_place_holder for x in example_list ]
split_arg = ' {} '.format(random_place_holder)
example_list = ' '.join(example_list).split(split_arg)
print(example_list)

老实说，如果您有一个不方便的占位符（例如 tobias_k 提到的），您最好使用任何其他解决方案。

决定计时： 使用：

example_list = ['This', 'is', None, 'an', None, 'example', 'list', None, '.'] * 10000
place_holder = None

我使用了一个更长的列表，这样随机字符串的创建就不是很耗时的部分，而且当你不使用大列表时，时间安排是愚蠢的。

这个解决方案： 每个循环 11.6 毫秒 ± 153 微秒（平均值 ± 标准偏差。7 次运行，每次 100 次循环）

Rakesh' 循环解决方案： 每个循环 25.8 毫秒 ± 819 微秒（平均值 ± 标准偏差。7 次运行，每次 10 次循环）

RoadRunner 的 groupby： 每个循环 34.4 毫秒 ± 1.21 毫秒（平均值 ± 标准偏差。7 次运行，每次 10 次循环）