Python / numpy：获取值列表的数组位置

发布时间：2020-12-20 12:20:04 所属栏目：Python 来源：网络整理

导读：假设我有一个值数组nameArr = [‘josh’,’is’,’a’,’person’]我想要一个像arrayLocation(nameArr,[‘a’,’is’])这样的函数来返回[2] 1]. 该功能是否已经存在,如果不存在,我该如何有效地实现它？解决方法使用numpy.where In [17]: nameArr = np.arra

假设我有一个值数组nameArr = [‘josh’,’is’,’a’,’person’]我想要一个像arrayLocation(nameArr,[‘a’,’is’])这样的函数来返回[2] 1].

该功能是否已经存在,如果不存在,我该如何有效地实现它？

解决方法

使用numpy.where

In [17]: nameArr = np.array(['josh','is','a','person'])

In [18]: [np.where(nameArr==i) for i in ['a','is']]
Out[18]: [(array([2]),),(array([1]),)]

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!