4

我正在尝试制作一个通用函数来减少这样的列表:

func(['a','b','c'],str.join) # --> ['a','b','c','ab','ac','bc','abc']

func(['a','b','c'],lambda: a,b:a+'x'+b) # --> ['a','b','c','axb','axc','bxc','axbxc']

我真的不知道该怎么做。我做了几次尝试,但没有一个成功。我很确定有一种方法可以用 reduce 来做到这一点,但我对使用这个功能不太满意。以下是一些尝试:

reduce(lambda a,b:[a,b,str(a)+str(b)],['a','b','c'])

reduce(str.join,['a','b','c'])

我想我在某处错过了递归。

我不是特别要求代码,欢迎任何帮助或建议。谢谢。

4

2 回答 2

3

itertools.combinations会给你一定长度的所有组合。我们为每个可能的子列表长度采用所有组合。然后,我们将您感兴趣的函数(lambda 函数,或本例中"x".join的 )映射到每个生成的组合。

>>> import itertools as it
>>> a = ['a','b','c']
>>> l = [map("x".join, list(it.combinations(a, l))) for l in range(1,len(a)+1)]
>>> l
[['a', 'b', 'c'], ['axb', 'axc', 'bxc'], ['axbxc']]

现在l是我们想要展平的列表列表:

>>> [ x for y in l for x in y]
['a', 'b', 'c', 'axb', 'axc', 'bxc', 'axbxc']
于 2013-06-17T08:35:25.203 回答
3

这个怎么样?

>>> import itertools
>>> def func(mylist, letter):
...     L = []
...     for i in range(len(mylist)):
...             L.append(list(itertools.combinations(mylist,i+1)))
...     return [letter.join(i) for i in itertools.chain.from_iterable(L)]
... 
>>> func(['a','b','c'], 'x')
['a', 'b', 'c', 'axb', 'axc', 'bxc', 'axbxc']
于 2013-06-17T08:36:35.443 回答