如何移除三维numpy数组中的第0列

问题示例 #

原三维NumPy数组：

import numpy as np

df = np.array([
    [[100,1,2,4,5,6,8],
     [87,1,6,20,22,23,34],
     [99,1,12,13,34,45,46]],
    [[64,1,10,14,29,32,33],
     [55,1,22,13,23,33,35],
     [66,1,6,7,8,9,10]],
    [[77,1,2,3,5,6,8],
     [811,1,2,5,6,8,10],
     [118,1,7,8,22,44,56]]
])

解决方法 #

利用NumPy的切片操作，直接在列维度上跳过索引为0的列，保留从索引1开始的所有列：

result = df[:, :, 1:]

预期结果 #

执行上述代码后，得到的数组为：

array([[[  1,   2,   4,   5,   6,   8],
        [  1,   6,  20,  22,  23,  34],
        [  1,  12,  13,  34,  45,  46]],

       [[  1,  10,  14,  29,  32,  33],
        [  1,  22,  13,  23,  33,  35],
        [  1,   6,   7,   8,   9,  10]],

       [[  1,   2,   3,   5,   6,   8],
        [  1,   2,   5,   6,   8,  10],
        [  1,   7,   8,  22,  44,  56]]])

内容的提问来源于stack exchange，提问作者user20250014