gpt4 book ai didi

计算所有产品组合的 Pythonic 方法

转载 作者:太空宇宙 更新时间:2023-11-04 03:56:11 26 4
gpt4 key购买 nike

假设我有一个素数列表 [ 2,3,5 ],我想获得所有 N^3 个此类产品的列表(或迭代器):

pow( 2, 0 ) * pow( 3, 0 ) * pow( 5, 0 ) 
pow( 2, 1 ) * pow( 3, 0 ) * pow( 5, 0 )
pow( 2, 0 ) * pow( 3, 1 ) * pow( 5, 0 )
pow( 2, 0 ) * pow( 3, 0 ) * pow( 5, 1 )
pow( 2, 1 ) * pow( 3, 1 ) * pow( 5, 0 )
pow( 2, 0 ) * pow( 3, 1 ) * pow( 5, 1 )
pow( 2, 1 ) * pow( 3, 0 ) * pow( 5, 1 )
[...]
pow( 2, N-1 ) * pow( 3, N-1 ) * pow( 5, N-1 )

执行此操作的 pythonic 方法是什么? (在长度为L的列表的情况下)

最佳答案

希望我没猜错。检查这个(N=3):

from itertools import product
from operators import mul

primes = [2,3,5]
n = 3

sets = product(*[[(i,j) for j in range(n)] for i in primes])
# Now 'sets' contains all the combinations you want. If you just wanted pow(i,j), write i**j instead and skip the map in the next enumeration
# list(sets)
#[((2, 0), (3, 0), (5, 0)),
# ((2, 0), (3, 0), (5, 1)),
# ((2, 0), (3, 0), (5, 2)),
# ... ... ...
# ((2, 2), (3, 2), (5, 0)),
# ((2, 2), (3, 2), (5, 1)),
# ((2, 2), (3, 2), (5, 2))]

productlist = []
for t in sets:
productlist.append(reduce(mul,map(lambda tp:tp[0]**tp[1],t)))

# now productlist contains the multiplication of each n(=3) items:
#[1, 5, 25, 3, 15, 75, 9, 45, 225, 2, 10, 50, 6, 30, 150, 18, 90, 450, 4, 20, 100, 12, 60, 300, 36, 180, 900]
# pow( 2, 0 ) * pow( 3, 0 ) * pow( 5, 0 ) = 1
# pow( 2, 0 ) * pow( 3, 0 ) * pow( 5, 1 ) = 5
# pow( 2, 0 ) * pow( 3, 0 ) * pow( 5, 2 ) = 25
# .... ...

或者一个衬垫可以是:

productlist = [reduce(mul,t) for t in product(*[[i**j for j in range(n)] for i in primes])]

关于计算所有产品组合的 Pythonic 方法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/18298953/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com