python - 查找不超过给定字符长度的所有字符串组合

Question

我有一个字符串列表，例如:['Lion','Rabbit','Sea Otter','Monkey','Eagle','Rat']

我正在尝试找出这些项目的可能组合总数，其中项目顺序很重要，以及当所有字符串用逗号分隔符连接小于给定长度时的总字符串长度。

因此，对于最大总字符串长度 14，我需要计算如下组合(并非详尽列表):

• 狮子
• 兔子
• 老鹰，狮子
• 狮子，鹰
• 狮子、鹰、鼠
• 鹰、狮、鼠
• 海獭，狮子
• 等...

但它不会包括总字符串长度超过 14 个字符限制的组合，例如 Sea Otter,Monkey

我知道对于这个非常有限的示例，手动计算或使用几个嵌套循环确定并不难，但实际用例将是一个包含数百个字符串的列表和更长的字符限制，这意味着手动编写的嵌套迭代次数会非常困惑......

我尝试通过 Python 的 itertools 来编写这个，但由于我发现的示例中没有一个足够接近我需要的，所以我一直迷路，尤其是在字符长度有限的情况下(不是 项目数量有限)以及允许在不同订单中重复组合的需要。

任何入门帮助都会很棒。

Answer 1

您可以在列表理解中使用 itertools.combinations 创建最多 3 个项目的列表(每个定义的更多项目将超过 14 个字符的组合)，同时按总字符串长度过滤:

import itertools

lst = ['Lion','Rabbit','Sea Otter','Monkey','Eagle','Rat']
sorted_lst = sorted(lst, key=len)

#find max number of items
for n, i in enumerate(sorted_lst):
    if len(','.join(sorted_lst[:n+1])) > 14:
        items_limit = n+1
        break

[x for l in range(1, items_limit) for x in itertools.combinations(lst, l) if len(','.join(x))<15]

附言。如果您需要排列(如示例输出)，请使用 itertools.permutations，您的问题是关于组合。