[英]Combinatorics in python numpy
我想写一个 function 来输出 2 个矩阵的所有可能组合:
def combine(*args):
return np.array(np.meshgrid(args)).T.reshape(-1, len(args)+1)
但是当通过时:
print(combine(np.array([1,2,3]), np.array([4,5,6])))
它输出:
[[1 2 3]
[4 5 6]]
我怎样才能让它发挥作用? 我想保持自动,而不是简单地传递 (args[0], args[1])
直接使用 python itertools:
In [134]: import itertools
In [135]: a,b = [1,2,3], [4,5,6]
In [137]: list(itertools.product(a,b))
Out[137]: [(1, 4), (1, 5), (1, 6), (2, 4), (2, 5), (2, 6), (3, 4), (3, 5), (3, 6)]
或者作为数组:
In [145]: np.array(list(itertools.product(a,b)))
Out[145]:
array([[1, 4],
[1, 5],
[1, 6],
[2, 4],
[2, 5],
[2, 6],
[3, 4],
[3, 5],
[3, 6]])
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.