numpy.delete删除一列或多列的方法

念桥上风景,为你痴守一世繁华。旅行的意义不在于浏览风景,而是到一个完全陌生的环境,放空自己,感受享受孤独!

基础介绍:

numpy.delete
numpy.delete(arr, obj, axis=None)[source]
 Return a new array with sub-arrays along an axis deleted. For a one dimensional array, this returns those entries not returned by arr[obj].
 Parameters:	
 arr : array_like
  Input array.
 obj : slice, int or array of ints
  Indicate which sub-arrays to remove.
 axis : int, optional
  The axis along which to delete the subarray defined by obj. If axis is None, obj is applied to the flattened array.
 Returns:	
 out : ndarray
  A copy of arr with the elements specified by obj removed. Note that delete does not occur in-place. If axis is None, out is a flattened array.

示例:

1.删除一列

>>> dataset=[[1,2,3],[2,3,4],[4,5,6]] 
>>> import numpy as np 
>>> dataset = np.delete(dataset, -1, axis=1) 
>>> dataset 
array([[1, 2], 
  [2, 3], 
  [4, 5]]) 

2.删除多列

arr = np.array([[1,2,3,4], [5,6,7,8], [9,10,11,12]]) 
np.delete(arr, [1,2], axis=1) 
array([[ 1, 4], 
  [ 5, 8], 
  [ 9, 12]]) 

以上这篇numpy.delete删除一列或多列的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

您可能有感兴趣的文章
使用Python向DataFrame中指定位置添加一列或多列的方法

python numpy中array与pandas的DataFrame转换方式

NumPy索引与切片的用法示例总结

Python多进程共享numpy 数组的方法

NumPy实现多维数组中的线性代数