 `y_train[y_train[:] == 1]` and `y_train[y_train[:] == 0]` both return 537 rows.
y_train[y_train[:] == 1]and
y_train[y_train[:] == 0]both return 537 rows.