对角列出嵌套列表中的元素

Question

这是我的第一篇文章,我是一个非常新的这个很棒的网站.

我想在对角线上列出3x3阵列中的所有元素:

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

预期产量:

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

另外,我如何将它推广到任何N x N阵列？

编辑:我注意到之前已经问过这个问题但是我正在寻找一种方法,而不需要导入numpy或任何库.

编辑2:我的一个同学提供了这个解决方案,我最喜欢它:

由于您从左下角开始查找对角线,您可以通过从方块的右上角开始并最终反转解决方案来完成此操作.我的方法是首先反转L中的每一行,然后将每个元素附加到其对应的对角线列表中.这里的见解是你开始在每一行K中追加元素,而不是在最终列表的第一个子列表中,但是从indice K开始.例如在将行[4,5,6]反转到行[6,5之后4],我会将6添加到我的对角排序列表的第二个子列表中(因为这是第2行),然后我将5添加到第3个子列表,然后我将4添加到第4个子列表.但是,如果我在对角线列表中没有第四个子列表,我会添加第四个空列表,然后用4填充它.

我的解释可能不太清楚,所以这里是我的代码.

def diagonal(l):

    L = l[:]
    return_list = [[] for i in range(len(L))]

    for line in range(len(L)):
        L[line].reverse()
        i = line

        for elem in L[line]:
            if i >= len(return_list):
                return_list.append([])

            return_list[i].append(elem)
            i += 1

    return_list.reverse()
    return return_list

Answer 1

仅使用Python(无NumPy):

import itertools as IT

L = [  [1, 2, 3],
       [4, 5, 6],
       [7, 8, 9] ]
N = len(L)
d = dict()
for i,j in IT.product(range(N), repeat=2):
    d.setdefault(j-i, []).append((i,j))

print([[L[i][j] for i,j in d[k]] for k in range(-N+1, N)])    
# [[7], [4, 8], [1, 5, 9], [2, 6], [3]]

或者,更好的是,使用Nemo的变换(推广到h x w形状矩阵:

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

h, w = len(L), len(L[0])

print([[L[h-1-q][p-q]
        for q in range(min(p, h-1), max(0, p-w+1)-1, -1)]
       for p in range(h+w-1)])                             
# [[7], [4, 8], [1, 5, 9], [2, 6], [3]]

我们还可以将此代码放在函数中以便于使用:

def diagonals(L):
    """
    https://stackoverflow.com/a/31373955/190597 (unutbu)
    >>> L = array([[ 0,  1,  2],
                   [ 3,  4,  5],
                   [ 6,  7,  8],
                   [ 9, 10, 11]])

    >>> diagonals(L)
    [[9], [6, 10], [3, 7, 11], [0, 4, 8], [1, 5], [2]]
    """
    h, w = len(L), len(L[0])
    return [[L[h - p + q - 1][q]
             for q in range(max(p-h+1, 0), min(p+1, w))]
            for p in range(h + w - 1)]


def antidiagonals(L):
    """
    >>> L = array([[ 0,  1,  2],
                   [ 3,  4,  5],
                   [ 6,  7,  8],
                   [ 9, 10, 11]])

    >>> antidiagonals(L)
    [[0], [3, 1], [6, 4, 2], [9, 7, 5], [10, 8], [11]]
    """
    h, w = len(L), len(L[0])
    return [[L[p - q][q]
             for q in range(max(p-h+1,0), min(p+1, w))]
            for p in range(h + w - 1)]