你好,我有三个Numpy数组如下所示。
>>> print A
[[ 1. 0. 0.]
[ 3. 0. 0.]
[ 5. 2. 0.]
[ 2. 0. 0.]
[ 1. 2. 1.]]
>>> print B
[[ 5. 9. 9.]
[ 37. 8. 9.]
[ 49. 8. 3.]
[ 3. 3. 1.]
[ 4. 4. 5.]]
>>>
>>> print C
[[ 0. 0. 0.]
[ 0. 6. 0.]
[ 1. 4. 6.]
[ 6. 2. 0.]
[ 0. 5. 4.]]
我希望将它们合并为
。
[[[ 1. 0. 0.]
[ 5. 9. 9.]
[ 0. 0. 0.]]
[[ 3. 0. 0.]
[ 37. 8. 9.]
[ 0. 6. 0.]]
[[ 5. 2. 0.]
[ 49. 8. 3.]
[ 1. 4. 6.]]
[[ 2. 0. 0.]
[ 3. 3. 1.]
[ 6. 2. 0.]]
[[ 1. 2. 1.]
[ 4. 4. 5.]
[ 0. 5. 4.]]]
我希望能从每个数组中选取一行。有没有简单的方法可以实现呢?我已经尝试过hstack
和vstack
,但它们并没有给出想要的结果。
谢谢!