python – 从数组数组中提取数组

2023年11月11日 182次阅读

我有这个数组：

arr = np.array([[[ -1.,  -1.,  -1.,  0.,   0.,   0.],
                [ 0.1,  0.1,  0.1,  2.,   3.,   4.]], # <-- this one

               [[ -1.,  -1.,  -1.,  0.,   0.,  -1.],
                [ 0.1,  0.1,  0.1, 16.,  17.,  0.1]], # <-- and this one

               [[ -1.,  -1.,  -1.,  0.,   0.,   0.],
                [ 0.1,  0.1,  0.1,  4.,   5.,   6.]], # <-- and this one

               [[  0.,   0.,   0., -1.,   0.,   0.],
                [  1.,   2.,   3., 0.1,   1.,   2.]], # <-- and this one

               [[ -1.,  -1.,   0.,  0.,   0.,   0.],
                [ 0.1,  0.1,   1.,  9.,  10.,  11.]]]) # <-- and the last one

我想在每个数组中提取第二个数组,结果如下：

res = [[ 0.1,  0.1,  0.1,  2.,   3.,   4.],
       [ 0.1,  0.1,  0.1, 16.,  17.,  0.1],
       [ 0.1,  0.1,  0.1,  4.,   5.,   6.],
       [  1.,   2.,   3., 0.1,   1.,   2.],
       [ 0.1,  0.1,   1.,  9.,  10.,  11.]]

我希望在一行代码中获得res,我试过这个,但它没有用

arr[:][1] # select the element 1 in each array
# I got
array([[ -1. ,  -1. ,  -1. ,   0. ,   0. ,  -1. ],
       [  0.1,   0.1,   0.1,  16. ,  17. ,   0.1]])

有谁能解释为什么？

我找到的唯一解决方案是明确指出每个索引(arr [0] [1] …),我不喜欢.

最佳答案 这是一个3D数组,您正在尝试选择第二个轴的第二个元素并沿其余轴提取所有元素.所以,它很简单 –

arr[:,1,:]

我们可以跳过列出：对于尾轴,所以它进一步简化为 –

arr[:,1]

样品运行 –

In [360]: arr
Out[360]: 
array([[[ -1. ,  -1. ,  -1. ,   0. ,   0. ,   0. ],
        [  0.1,   0.1,   0.1,   2. ,   3. ,   4. ]],

       [[ -1. ,  -1. ,  -1. ,   0. ,   0. ,  -1. ],
        [  0.1,   0.1,   0.1,  16. ,  17. ,   0.1]],

       [[ -1. ,  -1. ,  -1. ,   0. ,   0. ,   0. ],
        [  0.1,   0.1,   0.1,   4. ,   5. ,   6. ]],

       [[  0. ,   0. ,   0. ,  -1. ,   0. ,   0. ],
        [  1. ,   2. ,   3. ,   0.1,   1. ,   2. ]],

       [[ -1. ,  -1. ,   0. ,   0. ,   0. ,   0. ],
        [  0.1,   0.1,   1. ,   9. ,  10. ,  11. ]]])

In [361]: arr[:,1]
Out[361]: 
array([[  0.1,   0.1,   0.1,   2. ,   3. ,   4. ],
       [  0.1,   0.1,   0.1,  16. ,  17. ,   0.1],
       [  0.1,   0.1,   0.1,   4. ,   5. ,   6. ],
       [  1. ,   2. ,   3. ,   0.1,   1. ,   2. ],
       [  0.1,   0.1,   1. ,   9. ,  10. ,  11. ]])
点赞