столбец с разделенным массивом, даже если равное деление не может быть выполнено с использованием numpy

Question

Существует функция массива с разделением строк: array_split, которая:

Разбивает массив на несколько подмассивов одинакового или почти равного размера. Не вызывает исключение, если не может быть выполнено равное деление.

сейчас, я хочу иметь точно такое же функционально, как у этого array_split, но для столбцов.

Там есть клочокфункция: hsplit, которая выполняет:

Разбить массив на несколько под-массивов по горизонтали (по столбцам).

, но возвращает ошибку, если равное деление можетне может быть достигнуто.

Как это можно сделать?

Answer 1

Просто транспонируйте свой массив и используйте ту же функцию:

A = np.random.random((4,5))
print([x.T for x in np.array_split(A.T, 3)])