numpy.triu_indices#

numpy.triu_indices(n, k=0, m=None)[source]#

返回 (n, m) 数组上三角部分的索引。

参数:
nint

返回的索引有效的数组大小。

kint, 可选

对角线偏移（详见 triu）。

mint, 可选

返回的数组有效的数组列维度。默认情况下，m 等于 n。

返回:
indstuple, 形状为(2) 的 ndarrays, 形状为(n)

分别是行索引和列索引。行索引按非递减顺序排序，对应的列索引对于每行严格递增。

另请参阅

tril_indices

类似函数，用于下三角。

mask_indices

接受任意掩码函数的通用函数。

triu, tril

示例

>>> import numpy as np

计算两组不同的索引来访问 4x4 数组，一组用于从主对角线开始的上三角部分，另一组用于从主对角线右移两对角线开始的上三角部分。

>>> iu1 = np.triu_indices(4)
>>> iu1
(array([0, 0, 0, 0, 1, 1, 1, 2, 2, 3]), array([0, 1, 2, 3, 1, 2, 3, 2, 3, 3]))

注意，行索引（第一个数组）是非递减的，对应的列索引（第二个数组）对于每行严格递增。

以下是如何将它们与示例数组一起使用

>>> a = np.arange(16).reshape(4, 4)
>>> a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15]])

用于索引

>>> a[iu1]
array([ 0,  1,  2, ..., 10, 11, 15])

以及赋值

>>> a[iu1] = -1
>>> a
array([[-1, -1, -1, -1],
       [ 4, -1, -1, -1],
       [ 8,  9, -1, -1],
       [12, 13, 14, -1]])

这些只覆盖了整个数组的一小部分（主对角线右侧的两条对角线）

>>> iu2 = np.triu_indices(4, 2)
>>> a[iu2] = -10
>>> a
array([[ -1,  -1, -10, -10],
       [  4,  -1,  -1, -10],
       [  8,   9,  -1,  -1],
       [ 12,  13,  14,  -1]])
本页内容