Как эффективно перевернуть массив numpy по диагонали? - PullRequest
0 голосов
/ 14 апреля 2020

Допустим, у меня есть следующий массив (обратите внимание, что 1 в позиции [2,0] и 2 в позиции [3,4]):

[0, 0, 0, 0, 0]
[0, 0, 0, 0, 0]
[1, 0, 0, 0, 0]
[0, 0, 0, 0, 2]
[0, 0, 0, 0, 0]

и I хочу эффективно перевернуть его по диагонали так, чтобы:

[0, 0, 1, 0, 0]
[0, 0, 0, 0, 0]
[0, 0, 0, 0, 0]
[0, 0, 0, 0, 0]
[0, 0, 0, 2, 0]

Это не работает с fliplr, rot90 или flipud. Хотелось бы получить эффективный ответ, а не просто ответ, так как, к сожалению, это не выполняется с такими маленькими матрицами. Спасибо, Уилл.

1 Ответ

0 голосов
/ 14 апреля 2020

Я сумасшедший. np.rot90 (np.fliplr (x)) решает эту проблему.

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