Numpy Array Get Columns Pandas Disease