L oop по всем образцам Numpy ndarray

Question

Я пытаюсь l oop по всем выборкам моих данных; Вот как это выглядит и как это выглядит:

processed_data3
array([[[[1.81673904e-05, 5.00669221e-01],
         [1.65148740e-02, 5.52741534e-01],
         [1.55841024e-02, 4.39919449e-01],
         ...,
         [3.80455403e-03, 5.00042373e-01],
         [6.80686618e-01, 4.78582767e-01],
         [7.49290676e-04, 5.30804954e-01]],

processed_data3.shape
(100, 64, 256, 2)

Я хотел бы l oop по всем 100 выборкам данных и сделать то же самое, что я делаю только для один образец:

mask = np.zeros([64,256])
for i in range(64):
    for j in range(256):
        if processed_data3[0,i,j,0] >0.1:
            mask[i,j] = 1

В основном я хотел бы сохранить маску для каждого из этих 100 образцов в отдельном массиве масок, но я не уверен, как это сделать. Любые предложения, основанные на этом одном образце? Заранее большое спасибо!

Answer 1

Вы можете использовать np.where() для достижения желаемого результата. Никаких петель не требуется. Попробуйте это:

mask = np.where(processed_data3>0.1, 1, 0)[0][0]