gpt4 book ai didi

python - 获取 numpy 数组的 argsort

转载 作者:太空狗 更新时间:2023-10-30 01:14:32 25 4
gpt4 key购买 nike

我有一个 numpy 数组如下:

array([ True,  True,  True,  True,  True, False,  True,  True, False,
True, False, True, True, True, True, True, True, False,
False, False, False, False, True, True, False, False, False,
True, True, True, True, True, True, True, False, True,
True, True, True, False, True, True, False, False, True,
True, True, False, True, True, True, False],

我想获取所有 True 元素的索引。 numpy 中没有像 Pandas Series 那样的 get_loc 方法,同样也没有像列表那样的索引方法。我不想将其转换为列表然后使用 .index。

有什么想法吗?

最佳答案

使用 ndarray.nonzero :

>>> a.nonzero()
(array([ 0, 1, 2, 3, 4, 6, 7, 9, 11, 12, 13, 14, 15, 16, 22, 23, 27,
28, 29, 30, 31, 32, 33, 35, 36, 37, 38, 40, 41, 44, 45, 46, 48, 49,
50]),)

关于python - 获取 numpy 数组的 argsort,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/29956794/

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