python - reshape numpy 数组

Question

我有一个形状为 (20000,3) 的 numpy 数组 X:

X = array([[  3,   8,   4],
           [  1,   2,   4],
           ...
           [  5,   8,   4],
           [  3,   9,   4]])

我想删除最后一个“列”(即 4 所在的位置)，以获得如下内容:

X = array([[  3,   8],
           [  1,   2],
           ...
           [  5,   8],
           [  3,   9]])

我是否需要使用列表理解来执行此操作(即遍历数组元素，将结果保存到 list 并将 list 转换为 array，或者我可以使用内置的 numpy 函数，例如 extract 或 where？也许你推荐一些不同的东西？谢谢。

Answer 1

使用scipy 或numpy,指定数组，column_to_delete，行/列

scipy.delete(X, 2, 1)   #(np.array, index, axis)
numpy.delete(X, 2, 1) # would give the same result