給定一個 3Darr
np.random.seed(0)
arr=np.random.rand(4,3,3)
下面arr是一個
0.54881,0.71519,0.60276
0.54488,0.42365,0.64589
0.43759,0.89177,0.96366
0.38344,0.79173,0.52889
0.56804,0.92560,0.07104
0.08713,0.02022,0.83262
0.77816,0.87001,0.97862
0.79916,0.46148,0.78053
0.11827,0.63992,0.14335
0.94467,0.52185,0.41466
0.26456,0.77423,0.45615
0.56843,0.01879,0.61764
我想計算大于0.7最后兩個軸的總單元格。
哪個應該給出答案
3,3,5,2
從this OP提出的建議擴展，我認為這可以實作如下
[(darr>0.7).sum() for darr in arr]
但是，我想知道是否有另一種方法可以避免串列理解。
uj5u.com熱心網友回復：
您可以對多個軸求和。
(arr > 0.7).sum(axis=(1, 2))
array([3, 3, 5, 2])
轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/405783.html
標籤：
上一篇：根據情節選擇復制檔案