小编Vin*_*ent的帖子

从另一个数组中获取匹配项的索引

给定两个 np.arrays;

a = np.array([1, 6, 5, 3, 8, 345, 34, 6, 2, 867])
b = np.array([867, 8, 34, 75])
Run Code Online (Sandbox Code Playgroud)

我想得到一个与 b 具有相同维度的 np.array,其中每个值是 b 中的值出现在 a 中的索引,或者 np.nan 如果 b 中的值不存在于 a 中。

结果应该是;

[9, 4, 6, nan]
Run Code Online (Sandbox Code Playgroud)

a 和 b 将始终具有相同的维度数,但维度的大小可能不同。

就像是;

(伪代码)

c = np.where(b in a)
Run Code Online (Sandbox Code Playgroud)

但适用于数组(“in”不适用)

我更喜欢“单行”或至少是完全在阵列级别的解决方案,并且不需要循环。

谢谢!

python numpy

1
推荐指数
1
解决办法
62
查看次数

标签 统计

numpy ×1

python ×1