gpt4 book ai didi

python - 如何在另一个 ndarray 中找到一个 ndarray 的索引

转载 作者:行者123 更新时间:2023-12-04 15:11:44 27 4
gpt4 key购买 nike

<分区>

我有 a = np.arange(21).reshape(7,3)b = np.array([[3,4,5], [9 ,10,11]])

a = [[ 0  1  2]
[ 3 4 5]
[ 6 7 8]
[ 9 10 11]
[12 13 14]
[15 16 17]
[18 19 20]]

b = [[3 4 5]
[9 10 11]]

我想在 a 中找到 b 的行号。因此,我希望将 1 和 3 作为我的输出。我知道要找到索引,我可以使用 np.where()np.argwhere()。但是,我不知道我们是否可以使用它们来解决这个问题,或者我必须使用其他功能。我尝试了 c = np.argwhere(a == b) 但它给出了错误。

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