Iterate Through Columns In Numpy Which Function