Skip to main content
 首页 » 编程设计

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

2025年05月04日6arxive

<分区>

有一维值数组:

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)) 的掩码来创建它。