python中有关 numpy.ndarray 类似字典操作方法

NumPy字典操作技巧

最新推荐文章于 2023-09-23 21:40:55 发布

原创最新推荐文章于 2023-09-23 21:40:55 发布 · 6.3k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#numpy #ndarray #字典

python使用专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种使用NumPy的ndarray进行类似字典操作的方法，通过将两个数组作为键值对，实现快速查找。示例展示了如何利用数组间的比较和索引来实现这一过程。

numpy.ndarray 类似字典操作方法

假如，b和c是一一对应关系，则输入 a这个key值，可以通过b位置,对应的找到c位置

a = np.asarray(["8002"])
b = np.asarray(["8000","8001","8002","8003"])
c = np.asarray([1,2,3,4])

print (a == b)
print (c[a == b])

result:
[False False  True False]
[3]

a = np.asarray(["8000"])
b = np.asarray(["8000","8001","8002","8003"])
c = np.asarray([1,2,3,4])

print (a == b)
print (c[a == b])

result:
[ True False False False]
[1]