本文介绍了Python:从数组中提取索引值的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
Python: 我有一个具有3行和2列的阵列A2和具有1行和2列的A3。现在,我想打印A3中两个元素相对于A2的索引
示例
A2 = np.array([[0 6]
[15 1]
[1 15]])
A3 = [15 1]
现在,我要编写一个代码来打印A3的两个元素相对于A2的位置。
预期输出
(1,0)
请用python帮助我
我尝试使用
np.argwhere(A3 == A2).squeeze()
但这将输出为[1,0],[1,1]
推荐答案
您正在尝试查找数组%2的元素是否在数组%1中?
def search_two_elements_of_array_in_array(arr,arr2):
"""
:return: index if two elements of arr are in arr2 numpy array and print index
"""
for i in range(len(arr)):
if arr[i] in arr2:
print(i)
return i
return -1
这篇关于Python:从数组中提取索引值的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!
本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除!