Why does it print twice the “big” diagonal (matrix)

爷,独闯天下 提交于 2019-12-06 21:40:33

It appears that you have bad indentation in your print statement:

test_word = ''
for upper in range(len(matrix)):
    for rep1 in range(min(upper + 1, len(matrix[0]))):
        for rep2 in range(rep1, min(upper + 1, len(matrix[0]))):
            for j in range(rep1, rep2 + 1):
                test_word += (matrix[upper - j][j])
            print(test_word, end=' ')  # <--- indent left
            test_word = ''

But this however prints only:

P m mo o g gi giP i iP P n na nac naco a ac aco c co o a an anT anTs anTsP n nT nTs nTsP T Ts TsP s sP P a ag age ageA ageAr g ge geA geAr e eA eAr A Ar r

Not the 'porC, 'pdt', 'lo', 'e' diagonals.


This example prints all diagonals:

matrix = [
['P', 'o', 'P', 'o', 'P'],
['m', 'i', 'c', 's', 'r'],
['g', 'a', 'T', 'A', 'C'],
['n', 'n', 'e', 'r', 't'],
['a', 'g', 'o', 'd', 'o'],
['a', 'p', 'p', 'l', 'e']
]

def rotate(l, n):
    return l[n:] + l[:n]

def get_substrings(s):
    for j in range(0, len(s)):
        for i in range(j+1, len(s)+1):
            yield s[j:i]

def get_values(matrix):
    transposed = [*zip(*matrix)]
    for i in range(len(matrix[0])):
        transposed[i] = rotate(transposed[i], -i)
    transposed = [*zip(*transposed)]
    for i, lst in enumerate(transposed, 1):
        yield from get_substrings(''.join(lst[:i]))
    for i, lst in enumerate(transposed, 1):
        yield from get_substrings(''.join(lst[i:]))

for v in get_values(matrix):
    print(v, end=' ')

Prints:

P m mo o g gi giP i iP P n na nac naco a ac aco c co o a an anT anTs anTsP n nT nTs nTsP T Ts TsP s sP P a ag age ageA ageAr g ge geA geAr e eA eAr A Ar r p po por porC o or orC r rC C p pd pdt d dt t l lo o e

Edit: version without yield:

matrix = [
['P', 'o', 'P', 'o', 'P'],
['m', 'i', 'c', 's', 'r'],
['g', 'a', 'T', 'A', 'C'],
['n', 'n', 'e', 'r', 't'],
['a', 'g', 'o', 'd', 'o'],
['a', 'p', 'p', 'l', 'e']
]

def rotate(l, n):
    return l[n:] + l[:n]

def get_substrings(s):
    rv = []
    for j in range(0, len(s)):
        for i in range(j+1, len(s)+1):
            rv.append(s[j:i])
    return rv

def get_values(matrix):
    rv = []
    transposed = [*zip(*matrix)]
    for i in range(len(matrix[0])):
        transposed[i] = rotate(transposed[i], -i)
    transposed = [*zip(*transposed)]
    for i, lst in enumerate(transposed, 1):
        rv.extend(get_substrings(''.join(lst[:i])))
    for i, lst in enumerate(transposed, 1):
        rv.extend(get_substrings(''.join(lst[i:])))
    return rv

for v in get_values(matrix):
    print(v, end=' ')
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!