本文介绍了如何使某个整数的numpy数组下面的数字的所有数字组合的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！
问题描述
我有一个特定的numpy整数数组，我想创建一个数组数组，其中包含下面的所有数字组合。例如，[2,3]将产生
I have a certain numpy array of integers and I want to make an array of arrays that contains all combinations of numbers below it. As an example, [2,3] would yield
[[1,1],[2,1],[1,2],[2,2],[1,3],[3,3]]
因为这将是数字小于或等于[2,3]的数组的所有组合。
because that would be all the combinations of arrays with numbers less than or equal to [2,3].
推荐答案
使用 itertools.product ：
from itertools import product
a,b = 2,3
list(product(np.arange(a)+1,np.arange(b)+1))
输出：
[(1, 1), (1, 2), (1, 3), (2, 1), (2, 2), (2, 3)]
这篇关于如何使某个整数的numpy数组下面的数字的所有数字组合的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！