gpt4 book ai didi

python - 使用一维 bool 数组过滤二维数组

转载 作者:行者123 更新时间:2023-11-28 20:15:40 27 4
gpt4 key购买 nike

<分区>

我有两个 numpy 数组。

x = [[1,2], [3,4], [5,6]]
y = [True, False, True]

我想获取 X 的元素,其中 y 的对应元素为 True:

filtered_x = filter(x,y)
print(filtered_x) # [[1,2], [5,6]] should be shown.

我试过 np.extract,但它似乎只在 x 是一维数组时有效。如何提取 x 对应的 y 值为 True 的元素?

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