美文网首页
python实现matlab的nchoosek函数

python实现matlab的nchoosek函数

作者: 一路向后 | 来源:发表于2020-12-23 21:46 被阅读0次

1.源码实现

import itertools

def nchoosek(a, b, d=1, n=1):
        c = []
        for i in itertools.combinations(range(a,b+1,d),n):
                c.append(list(i))
        return c

a = nchoosek(2, 10, d=2, n=4)

print(a)

2.运行及结果

$ python3 example.py
[[2, 4, 6, 8], [2, 4, 6, 10], [2, 4, 8, 10], [2, 6, 8, 10], [4, 6, 8, 10]]

相关文章

网友评论

      本文标题:python实现matlab的nchoosek函数

      本文链接:https://www.haomeiwen.com/subject/hwounktx.html