【问题标题】:Find index of multiple elements in numpy array [duplicate]在numpy数组中查找多个元素的索引[重复]
【发布时间】:2018-01-08 23:41:45
【问题描述】:

给定一个整数数组,我需要找到存储在不同数组中的许多元素的索引。这是:

import numpy as np
a1 = np.array([ 4, 5, 6, 1, 2, 3, 7, 86, 9, 15])
a2 = np.array([ 2, 3, 5, 6, 9])

a1 是我的初始元素数组,a2 是包含我需要在a1 中为其索引的元素的数组。

在这种情况下,结果应该是:

a3 = ([4, 5, 1, 2, 8])

这似乎是一个相当简单的操作,但我一直无法弄清楚如何去做。

【问题讨论】:

  • 确实,这看起来像是那个问题的副本。对不起,我错过了。

标签: python arrays numpy


【解决方案1】:

你可以试试这个:

In [378]: (a1[:, None] == a2).argmax(axis=0)
Out[378]: array([4, 5, 1, 2, 8], dtype=int64)

【讨论】:

猜你喜欢
  • 2017-12-01
  • 1970-01-01
  • 2016-09-24
  • 2013-08-07
  • 2023-03-14
  • 2019-07-12
  • 1970-01-01
  • 1970-01-01
  • 2015-10-16
相关资源
最近更新 更多