python - Python中的对角网格遍历

为了清晰而编辑

我到处都在搜索这个，但还没有找到任何东西。我想遍历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']及其对应部分从左上方->底部至右。

最佳答案

出于某种推理，对于矩阵中的每个对角线，列号（或以下的x）和行号（y）之间的差是恒定的。我们可以将collections.defaultdict用作数据结构，该defaultdict被自动初始化为一个空列表，并且在所有矩阵元素上都有一个循环，找到每个元素所属的对角线，并在我们的getdiags中列出这些元素。

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指出其输入是等长字符串的列表，并且根据对角线列表来寻找答案。

我的上述getdiagonals与新的替代数据结构也能很好地工作，并且获得所需列表非常简单：

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

当然，这种转换也可以在defaultdict内实现，但这只是一个练习。

看，没有yield

# 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']

解决方案

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

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

关于python - Python中的对角网格遍历，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/34124583/