gpt4 book ai didi

python - 如何在 Python numpy 中的二维数组上应用一维掩码?

转载 作者:太空宇宙 更新时间:2023-11-04 06:15:05 24 4
gpt4 key购买 nike

<分区>

有一维值数组:

arr0 = numpy.array([8,0,9,5])

还有一个二维数组,其形状为(len(arr0),3):

arr1 = numpy.array([9,5,6],
[2,7,4],
[6,7,8],
[1,8,3])

我想创建一个 arr1 的掩码数组,如果 arr0[i] == 0,其中 arr1[i] 被掩码:

Result arr2 = [[9,5,6],
[-,-,-],
[6,7,8],
[1,8,3]]

创建这个新的掩码数组的优雅方法是什么?

我知道我可以使用形状为 (len(arr0),3) 的掩码来创建它。我希望我可以使用形状为 (len(arr0)) 的掩码来创建它。

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