numpy条件格式的元素

时间:2014-03-16 10:56:45

标签: python arrays numpy pyqtgraph

我有一个3D numpy数组,想要根据另一个元素的条件测试来更改特定元素。 (应用程序将更改RGBA图像数组的'alpha'以使用3D pyqtgraph图像中的透明度 - 理想情况下应该非常快)。

a= np.ones((2,4,5),dtype=np.int) #create a ones array
a[0,0,0] = 3 #change a few values
a[0,2,0] = 3
a[1,2,0] = 3
print(a)
>>>[[[3 1 1 1 1]
  [1 1 1 1 1]
  [3 1 1 1 1]]

 [[1 1 1 1 1]
  [1 1 1 1 1]
  [3 1 1 1 1]]]

现在我想有条件地测试最低维度的第一个元素(??),然后根据结果更改最后一个元素。

if a[0,:,:] > 1:   #this does not work - for example only - if first element > 1
    a[3,:,:]=255   #then change the last element in the same dimension to 255

print(a) #this is my idealized result
>>>[[[3 1 1 1 255]
  [1 1 1 1 1]
  [3 1 1 1 255]]

 [[1 1 1 1 1]
  [1 1 1 1 1]
  [3 1 1 1 255]]]

1 个答案:

答案 0 :(得分:2)

这似乎可以解决问题:

mask = a[:,:,0] > 1

a[:,:,4][mask] = 255

因此,索引只需稍微不同,然后它就是应用蒙版的标准做法。

修改的 @Ophion表示这写得好得多:

a[mask,:,-1] = 255