可以使用 numpy.delete() – 返回一个新数组,其中删除了一个轴的子数组。
import numpy as np a = np.array([1,2,3,4,5,6,7,8,9])
numpy.delete(a, index)
对于您的具体问题:
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]`
注意,numpy.delete()从array scalars开始返回一个新数组是不可变的,类似于Python中的字符串,所以每次对它进行更改时,都会创建一个新对象。
学会了删除numpy数组元素的小伙伴也可以分享给更多的人~更多Python学习推荐:PyThon学习网教学中心。
(推荐操作系统:windows7系统、Python 3.9.1,DELL G3电脑。)