преобразование одномерного массива numpy в квадратный массив n * n - PullRequest
0 голосов
/ 12 октября 2018

Я наткнулся на постановку задачи, в которой пользователь вводит данные в виде списка чисел.Общее количество входов неизвестно, но будет идеальный квадрат .Я должен преобразовать его в квадратную матрицу для выполнения дальнейших задач.Как я могу это сделать?Все, что я мог подумать, это:

my_list = [int(x) for x in input(prompt).split(" ")]
n= int(len(my_list)**0.5)  # no. of rows(or columns) for an nXn  matrix
my_arr = np.array(my_list).reshape([n,n])

Есть ли какие-либо встроенные функции или более эффективные методы, чем этот подход?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...