numpy删除特定行或列

numpy删除特定行或列

函数

numpy.delete(arr, obj, axis=None)

用法

• obj
  • slice, int or array of ints
  • Indicate indices of sub-arrays to remove along the specified axis
• axis
  • int, optional
  • axis = 0 删除选中行；axis = 1 删除选中列；axis = None 数组展开，flattened

例子

np.delete(arr, [1,2,4], 1)
删除数组arr的第2，3，5列