两个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)

6 声望

暂无个人描述~

Jason990420

1.9k 声望 / 個人 @ 個人

最佳答案

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]])

4年前评论

讨论数量: 1

Jason990420

1.9k 声望 / 個人 @ 個人

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]])

4年前评论

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

帮助