Question

我试图从Python中的矩阵中获取对角线，而不使用numpy（我实在无法使用它）。这里有人知道怎么做吗？

我想得到的例子：

get_diagonal ([[1,2,3,4],[5,6,7,8],[9,10,11,12]], 1, 1, 1)
Result: [1, 6, 11]

或者喜欢：

get_diagonal ([[1,2,3,4],[5,6,7,8],[9,10,11,12]], 1, 2, 1)
Result: [2, 7, 12]

直到知道我尝试了很多东西但是没有用。

def obter_diagonal(matrix, line, column, direc):
    d = []
    if direc == 1:
        for i in matrix:
            for j in i:
                if all(i == line, j == column):
                    d.extend(matrix[i][j])
    else:
        for i in matrix:
            for j in i:
                d.extend[len(matrix)-1-i][j]
    return d

如果direc==1我需要得到左边的对角线＆gt;右，顶部＆gt;底部。
如果direc==-1需要从右边获取diag＆gt; left，top-＆gt; bottom。

Answer 1

要获得领先的对角线，你可以做到

diag = [ mat[i][i] for i in range(len(mat)) ]

甚至

diag = [ row[i] for i,row in enumerate(mat) ]

为其他对角线玩类似的游戏。例如，对于反对角线（从右上角到左下角），您可以执行以下操作：

diag = [ row[-i-1] for i,row in enumerate(mat) ]

对于其他次要对角线，你必须在列表理解中使用if条件，例如：

diag = [ row[i+offset] for i,row in enumerate(mat) if 0 <= i+offset < len(row)]

Answer 2

def get_diagonal(m, i0, j0, d):
    return [m[(i0 + i - 1)%len(m)][(j0 + d*i - 1)%len(m[0])]
                for i in range(len(m))]

使对角线向前或向后移动：

m = [[1, 2, 3, 4],
     [5, 6, 7, 8],
     [9,10,11,12]]

print get_diagonal(m, 1, 1, 1)    # [1, 6, 11]
print get_diagonal(m, 1, 2, 1)    # [2, 7, 12]
print get_diagonal(m, 1, 4,-1)    # [4, 7, 10]

它甚至包裹在矩阵周围以获得对角线：

print get_diagonal(m, 1, 4, 1)    # [4, 5, 10]
print get_diagonal(m, 1, 1,-1)    # [1, 8, 11]
print get_diagonal(m, 3, 1, 1)    # [9, 2, 7 ]

Answer 3

对角线：

[m[i][i] for i in xrange(0, len(m))]

对于反对角线：

[m[i][~i] for i in xrange(0, len(m))]

Answer 4

好吧，我有一个适合我的解决方案。

输入：

第一行包含整数N

接下来的N行表示矩阵的行，每行包含描述列的以空格分隔的整数。

示例输入：

代码：

import sys


n = int(input().strip())
a = []
for a_i in range(n):
    a_t = [int(a_temp) for a_temp in input().strip().split(' ')]
    a.append(a_t)

pri_d = [];
pri_m = 0;
sec_d = [];
sec_m = n - 1;
for i in a:
    pri_d.append(i[pri_m]);
    sec_d.append(i[sec_m]);
    pri_m = pri_m + 1;
    sec_m = sec_m - 1;
print(pri_d);
print(sec_d);

输出：

[11, 5, -12]
[4, 5, 10]

Answer 5

由于这里没有人提到地图或lambdas，我会留下一个解决方案：

list(map(lambda x: x[a.index(x)], a))

那样在数组0，它将抓取元素0，依此类推。

对于相反的对角线，你可能想要自下而上翻转数组或者考虑数组的长度减一，然后减去当前的索引：

list(map(lambda x: x[(len(a) - 1) - a.index(x)], a)))

希望它有所帮助！

Answer 6

lis = []
lis2 = []
count = len(a)-1
count2 = 0
for i in range(len(a)):
   lis.append(a[count2][count])
   count-=1
   count2+=1
for i in range(len(a)):
   lis2.append(a[i][i])

print(lis,lis2)

这会打印左右对角线

不使用numpy获得对角线？

6 个答案: