Get row index of specific column in numpy array

I have written some code to find the maximum element of a specific element in python which I have denoted as column_max, and I now want to find the index for which row this is in but when I use np.where it only returns one index value.