在python列表中,如果我们想要删除一个或者连续几个元素,可以使用del()方法,在numpy数组,如果想要删除元素,可以使用numpy.delete()方法,但是numpy数组不支持删除数组元素,numpy.delete() 返回删除了某些元素的新数组。
1、np.delete()方法
numpy.delete()适用于numpy ndarray数组。
但是numpy数组不支持删除数组元素,numpy.delete() 返回删除了某些元素的新数组。
2、使用语法
numpy.delete(arr,obj,axis=None)
3、使用参数
arr:输入向量
obj:表明哪一个子向量应该被移除。可以为整数或一个int型的向量
axis:表明删除哪个轴的子向量,若默认,则返回一个被拉平的向量
4、返回值
返回一个新的数组,该数组具有沿删除的轴的子数组。
对于一维数组,这将返回arr[obj]未返回的那些条目。
5、使用实例
import numpy as np a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9]) index = [2, 3, 6] new_a = np.delete(a, index) print(new_a) #Prints `[1, 2, 5, 6, 8, 9]