我想写一个函数,输出两个矩阵的所有可能组合:
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])
