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

Answer 1

carnieri打败了我的numpy.triu_indices答案,但也有numpy.triu_indices_from一个数组作为输入而不是尺寸.

Answer 2

虽然索引位置的格式不同,但似乎你想要这个功能numpy.triu_indices.