2D numpy数组的上半对角线

Question

这似乎很简单（可以轻松编写三行循环），但是我如何使用numpy切片制作numpy数组上三角的索引位置列表？ 即：

给定一个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) ]

- Hooked

Answer 1

比我更快地回答了numpy.triu_indices的问题，但还有一个函数numpy.triu_indices_from，它接受一个数组作为输入而不是维度。

- Justin Peel

---

感谢Justin和Carnieri的回答。需要注意两件事，这需要numpy 1.4，而Ubuntu目前不是标准版本。此外，可以通过简单地解压缩答案即zip(*numpy.triu_indices(X))来获取所需的索引。 - Hooked

Answer 2

虽然索引位置的格式可能不同，但是你需要的函数似乎是numpy.triu_indices。

- carnieri

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)]

- naitsirhc

Answer 4

scipy.linalg.triu and nonzero

- Katriel