Как сохранить массив итераций в один? - PullRequest
0 голосов
/ 13 июля 2020

У меня есть код массива, как показано ниже, результатом является массив 2x2.

Как изменить его так, чтобы каждая итерация сохраняла один массив new_A (вместо 3 массивов)?

A = np. array([[1, 2],[3, 4]])
a = A.shape[0]
new_A = []
for i in range(1, a+2):
     B =mp( A, i)
     new_A += [B]
print(new_A)

Вот результат:

 [array([[1, 2],
       [3, 4]]), array([[ 7, 10],
       [15, 22]]), array([[ 37,  54],
       [ 81, 118]])]

1 Ответ

0 голосов
/ 13 июля 2020

Фактически у вас есть только один new_A . Произошло то, что ваша функция mp возвращает массив, поэтому вы вставляете 3 новых массива в свой new_A . Но у вас есть только один массив new_A. Возможно, вы хотите, чтобы ваша функция mp возвращала скалярное значение, а не массив.

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