python - 使用 numpy.tril 提取上方对角线处的列表元素

Question

嗨，我想使用 numpy.tril 提取上面对角线处的嵌套列表的元素。 。根据我的理解，设置参数k>0，它将返回上面的对角线。但是，我的代码没有返回预期的结果。

np.tril([[1,2,3],[4,5,6],[7,8,9]], 1)
>>> array([[1, 2, 0],
       [4, 5, 6],
       [7, 8, 9]])

预期输出:

array([1,2,3],
[4,5,0],
[7,0,0,])

Answer 1

您可以翻转数组，获取上三角形，然后将其翻转回来:

In [1]: import numpy as np

In [2]: a = np.array([[1,2,3],[4,5,6],[7,8,9]])

In [3]: np.triu(a[:, ::-1])[:, ::-1]
Out[3]:
array([[1, 2, 3],
       [4, 5, 0],
       [7, 0, 0]])