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列