python中使用np.delete()的实例方法

(编辑:jimmy 日期: 2024/11/11 浏览:2)

在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]