我有一個traind形狀為 的火車資料集(60000,28,28)。我想切出(60000)平均值為0.3或更大的所有樣本。我試過這個:
condition = (np.mean(traind[0:60000,:,:]) > 0.3)
subTraind = traind[condition]
但它引發了我跟隨錯誤:
IndexError: too many indices for array: array is 2-dimensional, but 3 were indexed
有沒有辦法用多個索引條件進行切片?
uj5u.com熱心網友回復:
np.mean默認情況下將計算整個陣列的平均值。您只想計算最后 2 個軸的平均值:
condition = traind.mean(axis=(1,2)) > 0.3
轉載請註明出處,本文鏈接:https://www.uj5u.com/qukuanlian/401277.html
