不用雙迴圈python如何統計numpy陣列中,下標x,y滿足條件的值數量?

時間 2021-06-06 17:49:21

1樓:彩雲上的棉花糖

對numpy中的array陣列進行索引時,不僅可以用整型、列表還可以用布林索引,你看看numpy的tutorial就知道了

2樓:henryPKU

可使用 Boolean Indexing:

>>>import

numpy

asnp

>>>k,

b,v=

1,1,

20>>>

arr=np.

reshape(np

.arange(5

*8),(

5,8))

>>>arrarray

([[0,1

,2,3

,4,5

,6,7

],[8,

9,10,

11,12,

13,14,

15],[16

,17,18

,19,20

,21,22

,23],[

24,25,

26,27,

28,29,

30,31],[32

,33,34

,35,36

,37,38

,39]])>>>

nrow

,ncol

=arr

.shape

>>>x,

y=np.

mgrid[0

:nrow,0

:ncol

]>>>

xarray

([[0,0

,0,0

,0,0

,0,0

],[1,

1,1,

1,1,

1,1,

1],[2

,2,2

,2,2

,2,2

,2],[

3,3,

3,3,

3,3,

3,3],

[4,4

,4,4

,4,4

,4,4

]])>>>

yarray

([[0,1

,2,3

,4,5

,6,7

],[0,

1,2,

3,4,

5,6,

7],[0

,1,2

,3,4

,5,6

,7],[

0,1,

2,3,

4,5,

6,7],

[0,1

,2,3

,4,5

,6,7

]])>>>

mask=(

y

x+b)

>>>mask

array

([[True

,False

,False

,False

,False

,False

,False

,False],[

True

,True

,False

,False

,False

,False

,False

,False],[

True

,True

,True

,False

,False

,False

,False

,False],[

True

,True

,True

,True

,False

,False

,False

,False],[

True

,True

,True

,True

,True

,False

,False

,False

]])>>>

res=

(arr

[mask

]

>>>resarray

([True

,True

,True

,True

,True

,True

,False

,False

,False

,False

,False

,False

,False

,False

,False

])>>>

res.

sum()6

3樓:曲奇

import

numpy

asnp

# 樣本資料:列分別為 x, y, value, 100行data=np

.random

.randn

(100,3

)# filter: k*x + b < y and value < vk=

0.5b

=0.1v=

0.1index=((

data

[:,0]*

k+b)

[:,1])&

(data

[:,2

]

filtered_data

=data

[index,:]

NBA季後賽採用雙迴圈賽制結果會如何?

紫川枕流 季後賽如此密集的賽程,還要球員東西部來回飛,競技狀態如何保持?之前回答類似問題,遇到乙個槓精,為了預防我特意說明一下。槓精 東西飛遠南北飛不也遠嗎?答 美國東西距離比南北長而且球隊分布在東西兩端的多,南北兩端的少。槓精 常規賽不也是東西部飛嗎?答 所以常規賽東西區球隊最多碰到兩次,而同半區...

如何評價復旦模聯2019全歐雙迴圈歷史聯動系統?

阿爾弗雷德之風 海戰IPC留點話吧。如此豪華的學團陣容 我甚至覺得我是最菜的 如此強大的能力 不少老師都是二次乃至多次共事了 我為我能在這FD歷史場奉獻力量感到無比的幸運,也對此盛景的難以重現而感到黯然神傷 不管如何這四天相信會給所有人留下一段特殊的記憶的,在MUN生涯的倒數時間裡,我會更加珍惜這些...

如何實現這個Python程式迴圈break?

兩株棗樹 大致我覺得可以把判斷退出迴圈的放在第乙個if語句,輸入3,直接break退出迴圈,後面的計算往後放。Python是按語句順序執行的,所以輸入3後,你的程式其實前面部分還是會執行,只有執行到最後乙個語句才會退出。 金剛腿十八 usr bin python3 coding utf 8 prin...