I have 2d array with 3 columns and N rows. In the third column there are only 0 or 1. I need to create 2 numpy arrays. They both contains first 2 columns of the given matrix, bu