numpy - 像字典一样使用数组

Question

我需要一个解决方案来使用像字典这样的数组。我不能使用字典，因为我使用 numba 并且我只能提交一个数组

corr_header = np.arange(0, 10).tolist()
corr_header_nr =  np.arange(10, 20).tolist()

corr_arr = list(zip(corr_header_nr, corr_header))
corr_arr

我可以在 corr_header_nr 中搜索一个值并从 corr_header 中获取该值。例如搜索 13 并得到 3 作为输出。感谢您的帮助

Answer 1

可以循环遍历数组，如果有匹配就返回。

def getoutput(key):
    return [k[1] for k in corr_arr if k[0]==key]

print(getoutput(13))