如何在Python中搜索Numpy数组中的数字

import numpy as np

# 创建一个Numpy数组
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])

# 查找数字5的位置
result = np.where(arr == 5)

print(result)  # 输出: (array([4], dtype=int64),)

# 查找在3到7之间的数字
result = arr[(arr >= 3) & (arr <= 7)]

print(result)  # 输出: [3 4 5 6 7]

# 检查数组中是否包含数字5和10
result = np.isin([5, 10], arr)

print(result)  # 输出: array([ True, False])

# 创建一个排序好的数组
sorted_arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])

# 查找数字5的插入点
result = np.searchsorted(sorted_arr, 5)

print(result)  # 输出: 4

index = np.where(arr == 5)[0]
print(index)  # 输出: [4]

result = arr[(arr >= 3) & (arr <= 7)]
print(result)  # 输出: [3 4 5 6 7]