python计算m的n次幂代码_用Python计算长度为M的第N序列

听起来你正在寻找itertools.product。我相信这会做你所要求的:

def seq(s):

length = 1

while True:

for p in itertools.product(s, repeat=length):

yield p

length += 1现在你可以做这样的事情:

>>> zip(range(10), seq(set((1, 2, 3))))

[(0, (1,)), (1, (2,)), (2, (3,)), (3, (1, 1)), (4, (1, 2)),

(5, (1, 3)), (6, (2, 1)), (7, (2, 2)), (8, (2, 3)), (9, (3, 1))]或这个:

>>> test_seq = itertools.izip(itertools.count(), seq(set((1, 2, 3))))

>>> for i in range(10):

... next(test_seq)

...

(0, (1,))

(1, (2,))

(2, (3,))

(3, (1, 1))

(4, (1, 2))

(5, (1, 3))

(6, (2, 1))

(7, (2, 2))

(8, (2, 3))

(9, (3, 1))这也可以使用其他itertools进一步压缩:

>>> from itertools import chain, product, count

>>> s = set((1, 2, 3))

>>> test_seq = chain.from_iterable(product(s, repeat=n) for n in count(1))

>>> zip(range(10), test_seq)

[(0, (1,)), (1, (2,)), (2, (3,)), (3, (1, 1)), (4, (1, 2)), (5, (1, 3)),

(6, (2, 1)), (7, (2, 2)), (8, (2, 3)), (9, (3, 1))]

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值