Верхняя диагональ двумерного массива

Question

Это кажется простым (и тривиально написать цикл из трех строк), но как я могу использовать нарезку на кусочки, чтобы составить список положений индекса верхней диагонали массива на кусках? * 1001 Т.е. *

Учитывая массив 4x4, я хотел бы, чтобы индексные местоположения были в X:

[ X X X X ]
[ 0 X X X ]
[ 0 0 X X ]
[ 0 0 0 X ]

Предоставление:

[ (0,0), (0,1), (0,2), (0,3), (1,1), (1,2), (1,3), (2,2), (2,3), (3,3) ]

Answer 1

Карниери опередил меня до ответа numpy.triu_indices, но есть также numpy.triu_indices_from, который принимает в качестве входных данных массив, а не размеры.

Answer 2

Хотя формат местоположений указателей отличается, похоже, вам нужна функция numpy.triu_indices.

Answer 3

Если вы работаете в Ubuntu и не хотите обновлять Numpy только для этого, вы можете использовать следующую функцию:

from itertools import chain
triu_indices = lambda x, y=0: zip(*list(chain(*[[(i, j) for j in range(i + y, x)] for i in range(x - y)])))

Пример:

In [26]: triu_indices = lambda x, y=0: zip(*list(chain(*[[(i, j) for j in range(i + y, x)] for i in range(x - y)])))

In [27]: triu_indices(4)
Out[27]: [(0, 0, 0, 0, 1, 1, 1, 2, 2, 3), (0, 1, 2, 3, 1, 2, 3, 2, 3, 3)]

In [28]: zip(*triu_indices(4))
Out[28]: 
[(0, 0),
 (0, 1),
 (0, 2),
 (0, 3),
 (1, 1),
 (1, 2),
 (1, 3),
 (2, 2),
 (2, 3),
 (3, 3)]

Answer 4

scipy.linalg.triu и nonzero