3

numpy/scipy 中是否有一个函数可以返回方阵的一个三角形（上或下）的所有元素？

例如：

matrix = [[1,  2,  3],
          [4,  5,  6],
          [7,  8,  9]]

三角形（上和下）：

up = [1,2,3,5,6,9]
down = [1,4,5,7,8,9]

或者

up = [1,2,5,3,6,9]
down = [1,4,7,5,8,9]

谢谢！

编辑：

是的，有两个函数可以帮助您做到这一点：（np.triu_indices(n)对于上三角）和np.tril_indices(n)（对于下三角）。

谢谢卡特里拉克斯！

4

2 回答 2

9

元素的顺序重要吗？

通常，矩阵的上下三角形是围绕另一个对角线（左上角到右下角）取的。要解决这个问题，您需要在两条对角线之间“翻转”，您可以使用np.fliplr(matrix). 这将为您提供正确的元素，但以“自然”顺序（逐行，每一行变得更短）。np.flipud您还可以通过翻转另一种方式 ( )来获得逐列的顺序。但是我不知道有什么方法可以得到你正在使用的“按较小的对角线读取”顺序，而不是一次读取一个对角线的矩阵。

要获取对角线元素，您可以使用np.triu_indices（或者，对于下三角形，np.tril_indices）获取它们的索引，然后按它们索引。

>>> matrix
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])
>>> np.fliplr(matrix)[np.triu_indices(3)]
array([3, 2, 1, 5, 4, 7])
于 2012-05-31T19:27:08.310 回答
2

以下代码提取包括对角线的下三角形：

hstack(mat[i][:i+1] for i in xrange(mat.shape[0]))

这将提取没有对角线的下三角形：

hstack(mat[i][:i] for i in xrange(mat.shape[0]))
于 2014-05-28T13:52:06.147 回答