如何删除 numpy 数组中的列？

假设我们有一个 5x4 矩阵。 我们只需要删除第一个维度。 我们怎样才能做到这一点numpy?

array([[  0.,   1.,   2.,   3.],
       [  4.,   5.,   6.,   7.],
       [  8.,   9.,  10.,  11.],
       [ 12.,  13.,  14.,  15.],
       [ 16.,  17.,  18.,  19.]], dtype=float32)

I tried:

arr = np.arange(20, dtype=np.float32)
matrix = arr.reshape(5, 4)
new_arr = numpy.delete(matrix, matrix[:,0])
trimmed_matrix = new_arr.reshape(5, 3)

它看起来有点笨重。 我做得正确吗？ 如果是，是否有更干净的方法来删除尺寸而不重塑？

如果你想从 2D Numpy 数组中删除一列，你可以像这样指定列

保留所有行并删除第 0 列（或从第 1 列开始直到最后）

a[:,1:]

另一种方法是您可以指定要保留的列（如果您愿意，还可以更改顺序） 这将保留所有行并且仅使用列 0,2,3

a[:,[0,2,3]]

可以找到这方面的文档here http://docs.scipy.org/doc/numpy-1.10.1/user/basics.indexing.html

如果你想要专门删除列的东西，你可以这样做：

idxs = list.range(4)
idxs.pop(2) #this removes elements from the list
a[:, idxs]

和 @hpaulj 提出了 numpy.delete()

这将是如何返回 'a' 的视图，其中沿 axis=1 删除了 2 列（0 和 2）。

np.delete(a,[0,2],1)

这实际上并没有从“a”中删除项目，它的返回值是一个新的 numpy 数组。