numpy怎么删除指定元素

1205
2024/2/7 14:32:40
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

使用NumPy库中的delete()函数可以删除指定的元素。delete()函数的用法如下:

numpy.delete(arr, obj, axis=None)

其中,arr是要操作的数组,obj是要删除的元素的索引或切片对象,axis是要删除的轴。如果axis未指定,则将数组展开为一维数组,然后删除指定的元素。

下面是一个示例,演示了如何使用delete()函数删除指定的元素:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
new_arr = np.delete(arr, 2)  # 删除索引为2的元素
print(new_arr)  # 输出: [1 2 4 5]

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
new_arr = np.delete(arr, 1, axis=0)  # 删除第1行
print(new_arr)
# 输出:
# [[1 2 3]
#  [7 8 9]]

new_arr = np.delete(arr, 1, axis=1)  # 删除第1列
print(new_arr)
# 输出:
# [[1 3]
#  [4 6]
#  [7 9]]

在以上示例中,np.delete()函数分别删除了一维数组和二维数组中的指定元素。

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: NumPy怎么与MPI集成使用