python-3.x - 如何获得字符串(元音或辅音)的组合？

Question

给定索引的可能组合数例如单词 BANANA 的索引 [0]应该给我:

{'B',
 'BA',
 'BAN',
 'BANA',
 'BANAN',
 'BANANA'}

word='BANANA'
indices=[0,2,4]
def find_combinations(word:str,indices:list):
    a=[''.join(l) for i in range(len(word)) for l in combinations(word, i+1)]
    b=[x for x in a if x.startswith('B')]
    return b

输出:

set(b)
{'B',
 'BA',
 'BAA',
 'BAAA',
 'BAAN',
 'BAANA',
 'BAN',
 'BANA',
 'BANAA',
 'BANAN',
 'BANANA',
 'BANN',
 'BANNA',
 'BN',
 'BNA',
 'BNAA',
 'BNAN',
 'BNANA',
 'BNN',
 'BNNA'}

期望的输出:

{'B',
 'BA',
 'BAN',
 'BANA',
 'BANAN',
 'BANANA'}

Answer 1

您不需要组合，您可以使用 slices 轻松生成从特定索引开始的前缀。和范围。

from typing import List

def get_parts(word: str, start: int) -> List[str]:
    return [word[start:i] for i in range(start + 1, len(word) + 1)]

(显然，如果您需要它是一个set，您可以更改为return { ... }。

>>> get_parts("BANANA", 0)
['B', 'BA', 'BAN', 'BANA', 'BANAN', 'BANANA']

>>> get_parts("BANANA", 2)
['N', 'NA', 'NAN', 'NANA']

>>> get_parts("BANANA", 4)
['N', 'NA']