使用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()
函數分別刪除了一維數組和二維數組中的指定元素。