Найти продукт любого подмножества элементов из списка
Каков наилучший способ найти произведение любого количества элементов из списка?
Например, если у меня есть [a,b,c]
как вход, я должен получить [a,b,c,a*b,a*c,b*c,a*b*c]
в качестве вывода (порядок элементов для вывода не имеет значения.)
PS: Мы можем сделать это рекурсивно? (например, вам нужен продукт a*b
и c
, чтобы получить продукт a*b*c
.
Любая идея или предложение приветствуются. Спасибо заранее!
Ответы
Ответ 1
Здесь вы идете:
from itertools import combinations
l = [2, 3, 5]
result = []
for i in range(1, len(l) + 1):
result += list(combinations(l, i))
multiplied_result = [reduce(lambda x, y: x*y, lst) for lst in result]
Теперь, если мы напечатаем результат, получим
>>> print listmap
[2, 3, 5, 6, 10, 15, 30]
Ответ 2
Вы можете использовать itertools.combinations
в понимании списка:
>>> def find_mul(li):
... return [[reduce(lambda x,y:x*y,j) for j in combinations(li,i)] for i in xrange(2,len(li)+1)]
...
DEMO:
>>> [list(combinations([2,3,4],i)) for i in xrange(2,len([2,3,4])+1)]
[[(2, 3), (2, 4), (3, 4)], [(2, 3, 4)]]
>>> l=[2,3,4]
>>> find_mul(l)
[[6, 8, 12], [24]]