基于滤波器2D递增无循环的数组索引

问题描述

我有一个2D数组,我需要根据另一个数组中的标志来增加索引。递增的索引可以重复。我可以使用循环,但我想提高效率。

一维示例:

a = np.zeros(10,dtype=np.int)
iflag = [1,3,5,1,7,1]
a[iflag] =+1 

给予 一种 数组([0,1,0,1,0,1,0,1,0,0])

我想要的是 一种 数组([0,6,2,0])

我需要在2D模式下完成

array = np.zeros(400,600)
flag.shape 
 (2000,20)
for i in np.range(20):
  for j in np.range(2000) :
     if (flag[i,j] ==1) : 
# ilat,ilon and flag are all same shape
# ilat and ilon will contain positive int that are indices to array    
        array[ilat[i,j],ilon[i,j]] += 1

我想做的事情是这样的:

  iflag= np.where(flag ==1)
  array[ilat[iflag],ilon[iflag]] += 1

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)