gpt4 book ai didi

python - 在另一个数组中查找数组的字符串元素

转载 作者:行者123 更新时间:2023-12-04 09:23:17 27 4
gpt4 key购买 nike

我有两个排序数组 hM :

h = np.array(['blue', 'red', 'white'])
M = np.array(['blue', 'green', 'orange', 'red', 'white'])
并想找到 h 的每个元素所在的索引出现在 M我可以吗 np.where为了这?
此外,可能是 h 的元素可能不会出现在 M在这种情况下,我不需要它的索引。元素不重复。

最佳答案

这是另一种方式,是一个单行,将返回 ndarray比赛的索引。

np.nonzero(np.in1d(M, h))[0]
输出
array([0, 3, 4])

关于python - 在另一个数组中查找数组的字符串元素,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63064571/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com