获取两个数组对应位置上元素值相等的位置信息，题目详情见正文。_CDA答疑社区

热线电话：13121318867

登录

yuechuchen

2020-04-27 阅读量: 758

获取两个数组对应位置上元素值相等的位置信息，题目详情见正文。

有数组a = np.array([4,5,1,2,3,2,1,5])和b = np.array([4,5,2,6,9,2,0,5])，获取对应位置上元素值相等的位置信息：实现效果如下：array([0, 1, 5, 7])，注意array([0, 1, 5, 7])里的元素全部都是关于数组a和数组b中元素的位置信息

解题思路：

判断对应位置元素是否一致，只需要让两个数组相减，结果为0的位置上的元素即为元素值一致，然后获取整个数组所有位置信息，和非0的位置信息求差集即可

具体代码如下：

a = np.array([4,5,1,2,3,2,1,5])

b = np.array([4,5,2,6,9,2,0,5])

np.setdiff1d(np.array(range(len(a))),np.nonzero(a-b)) #np.nonezero()得到非0元素的位置信息，与所有元素的位置信息求差集

两个列表在【0，1，5，7】位置上的元素值相等

1.3161

1

2

关注作者

收藏

评论(0)

发表评论

暂无数据

CDA考试动态

CDA报考指南

推荐帖子