gpt4 book ai didi

python - 如何使用另一个数组的值过滤 numpy 数组?

转载 作者:IT老高 更新时间:2023-10-28 21:00:29 26 4
gpt4 key购买 nike

我有两个 NumPy 数组,例如:

a = [1,2,3,4,5]

和一个过滤器数组,例如:

f = [False, True, False, False, True]

len(a) == len(f)

如何获得一个新的 numpy 数组,其中只有 f 中的相同索引为 True 的值?在我的例子中:[2, 5].

根据接受的解决方案(具有不同的值):

>>> a = numpy.array([ 1,  2,  3,  4,  5,  6,  7,  8,  9, 10])
>>> b = numpy.array([True, False, True, False, True, False, True, False, True, False])
>>> a[b]
array([1, 3, 5, 7, 9])

最佳答案

NumPy 支持 boolean indexing

a[f]

这假设 af 是 NumPy 数组而不是 Python 列表(如问题中所示)。您可以使用 f = np.array(f) 进行转换。

关于python - 如何使用另一个数组的值过滤 numpy 数组?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/9296658/

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