numpy数组删除元素
可以使用numpy.delete()函数来删除numpy数组中的元素。该函数的语法如下:
numpy.delete(arr, obj, axis=None)
其中,arr代表要删除元素的数组,obj代表要删除的元素的索引或切片,axis代表要删除元素的轴,默认为None,表示将数组展平后删除。
示例:
import numpy as np
# 创建一个数组
arr = np.array([1, 2, 3, 4, 5])
# 删除第二个元素
new_arr = np.delete(arr, 1)
print(new_arr) # [1 3 4 5]
# 创建一个二维数组
arr = np.array([[1, 2], [3, 4], [5, 6]])
# 删除第二行
new_arr = np.delete(arr, 1, axis=0)
print(new_arr) # [[1 2]
# [5 6]]
# 删除第二列
new_arr = np.delete(arr, 1, axis=1)
print(new_arr) # [[1]
# [3]
# [5]]
注意:numpy.delete()函数返回的是一个新的数组,原数组并未改变。
原文地址: https://www.cveoy.top/t/topic/bwTm 著作权归作者所有。请勿转载和采集!