Блог

Я новичок в python/ программировании. Я пытаюсь создать небольшую функцию, в которой циклы в каждой строке матрицы x на 3 содержат ошибки. Я хочу сохранить все строки без ошибок в новой матрице x на 3. Например, входные данные могут быть:

 [[50.     0.096  2.   ]
 [15.     0.517  3.   ]
 [35.     1.086  4.   ]
 [40.     0.934  2.   ]
 [35.     0.109  1.   ]]
 

Мой код:

 def function(f5):
    for i in range(f5.shape[0]):
        if f5[i,0] < 10:
            print("error1")
        elif f5[i,0] > 60: 
            print("error2")
        elif f5[i, 1] < 0:
            print("error3")
        elif (f5[i, 2] != 1) and (f5[i, 2] != 2) and (f5[i, 2] != 3) and (f5[i, 2] != 4):
            print("error4")
        else:
            #I'm not sure how to save the rows that pass these tests in a 
             new matrix.
 

Я был бы ОЧЕНЬ признателен за любую помощь.