两个numpy数组拼接问题请教

有如下两个numpy数组a和b需要拼接为如下结果:
[[0,11],
[0,12],
[1,11],
[1,12],
[2,11],
[3,11]]
请问如何才能优雅实现

import numpy as np
a = np.arange(3)
b = np.arange(11,13)
Jason990420
最佳答案

Try this,

from itertools import product
import numpy as np

a = np.arange(3)
b = np.arange(11,13)
c = np.array(list(product(a, b)))
>>> c
array([[ 0, 11],
       [ 0, 12],
       [ 1, 11],
       [ 1, 12],
       [ 2, 11],
       [ 2, 12]])
2年前 评论
讨论数量: 1
Jason990420

Try this,

from itertools import product
import numpy as np

a = np.arange(3)
b = np.arange(11,13)
c = np.array(list(product(a, b)))
>>> c
array([[ 0, 11],
       [ 0, 12],
       [ 1, 11],
       [ 1, 12],
       [ 2, 11],
       [ 2, 12]])
2年前 评论

讨论应以学习和精进为目的。请勿发布不友善或者负能量的内容,与人为善,比聪明更重要!