gpt4 book ai didi

python - np.where 相当于多维 numpy 数组

转载 作者:行者123 更新时间:2023-11-28 20:24:02 25 4
gpt4 key购买 nike

假设您有一个 numpy 数组作为 array([[5],[1,2],[5,6,7],[5],[5]])。是否有一个函数,例如 np.where,可用于返回所有行索引,其中 [5] 是行值?例如,在上面的数组中,返回值应该是 [0, 3, 4] 表示 [5] 行号。

请注意,数组中的每一行的长度都可以不同。

谢谢大家,你们都应该得到最佳答案,但我给第一个打了绿色标记:)

最佳答案

应该这样做:

[i[0] for i,v in np.ndenumerate(ar) if v == [5]]
=> [0, 3, 4]

关于python - np.where 相当于多维 numpy 数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/18029710/

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