python-元素的numpy条件格式

2019-11-22 01:08:33 阅读：296 来源： 互联网

我有一个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]]]

解决方法:

这似乎可以解决问题：

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

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

因此,索引只需稍有不同,这只是应用蒙版的标准做法.

编辑
@Ophion显示这更好地写为：

a[mask,:,-1] = 255

标签：pyqtgraph,arrays,python,numpy
来源： https://codeday.me/bug/20191122/2056024.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

python-元素的numpy条件格式