给定两个 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”不适用)
我更喜欢“单行”或至少是完全在阵列级别的解决方案,并且不需要循环。
谢谢!