TENGO UNA MATRIZ DE 100 X 3, donde la columna 3 son datos de tipo binario (1 y 0) Segrega los datos en dos matrices diferentes segun el índice de la tercera columna

def segregar(matriz):
  filas_uno = []
  filas_cero = []
  for i in range(len(matriz)):
    if matriz[i][2] == 1:
      filas_uno.append(matriz[i])
    else:
      filas_cero.append(matriz[i])
  return filas_uno, filas_cero