A.argmax(axis = 0) возвращает список позиций максимальных элементов в каждом столбце, поэтому может удалить от одного столбца (если все максимальные элементы в одной строке) до всех столбцов (если номера позиций максимальных элементов не совпадают).

A.max(axis = 0).argmax() возвращает номер столбца с максимальным элементом.

Mutant
И почему 3 параметр = 1, разве 1 это не по строчкам удаление должно быть, а о по столбцам?

По видимому, удаляет не сам столбец, а соответствующие элементы в строках (IMHO).