Python Numpy Array Get Columns From Dataframe