如何一次性获取NumPy数组中特定元素的索引？
- 内容介绍
- 文章标签
- 相关推荐
本文共计428个文字，预计阅读时间需要2分钟。
要找到数组 `a` 中每个元素在数组 `b` 中的索引，可以使用 `numpy` 库中的 `searchsorted` 函数。以下是一个简化的代码示例：
pythonimport numpy as np
a=np.array([1, 2, 4])b=np.array([1, 2, 3, 10, 4])
使用searchsorted找到索引indices=np.searchsorted(b, a, side='left')
print(indices)
这段代码将输出每个元素在数组 `b` 中的索引，即 `[0, 1, 4]`。
本文共计428个文字，预计阅读时间需要2分钟。
要找到数组 `a` 中每个元素在数组 `b` 中的索引，可以使用 `numpy` 库中的 `searchsorted` 函数。以下是一个简化的代码示例：
pythonimport numpy as np
a=np.array([1, 2, 4])b=np.array([1, 2, 3, 10, 4])
使用searchsorted找到索引indices=np.searchsorted(b, a, side='left')
print(indices)
这段代码将输出每个元素在数组 `b` 中的索引，即 `[0, 1, 4]`。