Question

为了清晰而编辑

我到处寻找这个，但没找到任何东西。我想遍历一个2D字符串列表，并像在这里一样显示切片 Traverse 2D Array (Matrix) Diagonally 但是在python中。

lines = ['xmfycxvc',
         'caubmekv',
         'awactivb',
         'paphzkcn',
         'sbsaakjy',
         'tsewlhvk']
diagonals = []
i = 0
while i < len(lines):
   j = 0
   diagonal = ''
   while j < len(lines[0]):
      diagonal += lines[j][j]
      i += 1
diagonals.append(diagonal)
print(diagonals)

我知道我的索引是错的，但我已经尝试了所有内容，但仍然无法使其像链接一样。我最接近的是拥有每个对角线，但是它也会围绕矩阵包围像['xaahah','muczkv','fbtkjk','cevnss','xkbpbs','vvaasw','xxwpal']之类的球体，但我不想要它。

我想对角穿过琴弦矩阵并打印对角线，例如['x','cm','aaf','pwuy','saabc','tbpcmx','sshtev','eazikc','wakvv','lkcb','hjn','vy','k']及其对应部分从左上角开始 - ＆gt; botom右

Answer 1

从一些推理我们得到，对于矩阵中的每个对角线，列号（或后面的x）和行号（y）之间的差异是不变的。我们可以使用自我初始化为空列表的collections.defaultdict作为我们的数据结构，并在所有矩阵元素上有一个循环，找到每个元素所属的对角线，并在defaultdict中列出这些元素。

def getdiags(matrix, nr, nc):
    from collections import defaultdict
    d = defaultdict(list)
    for y in range(nr):
        for x in range(nc):
            d[x-y].append(matrix[y][x])
    return d

我们也可以从效用函数中获益，以便我们的结果出现

def printdiags(diags):
    for i in sorted(diags):
        print diags[i]

最后我们在IPython中测试我们的东西

In [1]: def getdiags(matrix, nr, nc):
        from collections import defaultdict
        d = defaultdict(list)
        for y in range(nr):
                for x in range(nc):
                        d[x-y].append(matrix[y][x])
        return d
   ...:     

In [2]: def printdiags(diags):
        for i in sorted(diags):
                print diags[i]
   ...:         

In [3]: from pprint import pprint

In [4]: m = [["%3.3d"%(r*100+c)  for c in range(5)] for r in range(4)]

In [5]: pprint(m)
[['000', '001', '002', '003', '004'],
 ['100', '101', '102', '103', '104'],
 ['200', '201', '202', '203', '204'],
 ['300', '301', '302', '303', '304']]

In [6]: diags = getdiags(m, 4, 5)

In [7]: printdiags(diags)
['300']
['200', '301']
['100', '201', '302']
['000', '101', '202', '303']
['001', '102', '203', '304']
['002', '103', '204']
['003', '104']
['004']

In [8]:

全部

附录

在后期编辑中，OP表示他们的输入是一个相等长度的字符串列表，并且根据对角线列表寻找答案。

上面我的getdiags也适用于新的替代数据结构，获取搜索列表非常简单：

def listofdiags(diags):
    return [''.join(diags[i]) for i in sorted(diags)]

当然，此转换也可以在 getdiagonals内实施，但这仍然是一项练习。

看，没有defaultdict

# List of Strings from Diagonals def lsd(m, nr, nc): res = ["" for i in range(nr+nc-1)] for r in range(nr): for c in range(nc): i = c-r+nr-1 res[i] = res[i]+m[r][c] return res pprint(lsd(m, 4, 5)) # ['300', # '200301', # '100201302', # '000101202303', # '001102203304', # '002103204', # '003104', # '004']

yield解决方案

以下效率较低，但为了完整起见，这里是：

def enumdiags(m, nr, nc): for i in range(nr+nc-1): s = "" for r in range(nr): for c in range(nc): if c-r+nr-1 == i : s = s+m[r][c] yield i, s for i, s in enumdiags(m, 4, 5): print i, s # 0 300 # 1 200301 # 2 100201302 # 3 000101202303 # 4 001102203304 # 5 002103204 # 6 003104 # 7 004

Answer 2

已编辑以降低复杂性

    def printer(matrix, i, j, listOfDiagonals):
        if i + 1 > len(matrix) or j + 1 > len(matrix[0]):
            return listOfDiagonals
        else:
            listOfDiagonals.append(matrix[i][j])
            printer(matrix, i+1, j+1, listOfDiagonals)

    matrix = [[0 for i in range(10)] for j in range(10)]
    for i in matrix:
        print i
    list = []
    printer(matrix, 0, 0, list)
    print list

矩阵是你的矩阵，然后i和j是你的索引。

这将递归地将索引位置处的内容添加到列表中，该列表最终在到达矩阵的边界时返回。这个设计的好处是矩阵不需要是正方形的，它仍然可以穿过对角线。

Python中的对角网格遍历

2 个答案:

附录

看，没有`defaultdict`

`yield`解决方案

Python中的对角网格遍历

2 个答案:

附录

看，没有defaultdict

yield解决方案

看，没有`defaultdict`

`yield`解决方案