我有一个形状为 (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？也许你推荐一些不同的东西？谢谢。
请您参考如下方法:
使用scipy 或numpy, 指定数组，column_to_delete，行/列
scipy.delete(X, 2, 1) #(np.array, index, axis)
numpy.delete(X, 2, 1) # would give the same result
