蓝桥杯python：寻找2020

题目：

程序说明：
打开文件以后，将文件一行一行的读取存入列表，返回的是一个字符串对象，再分别对横向进行寻找，纵向进行寻找，斜向进行寻找即可。
全部代码：

li = []
with open(r'2020.txt', 'r') as fp:
    for line in fp.readlines():
        li.append(line.strip())
m = len(li)
n = len(li[0])
num = 0
for i in range(m):
    for j in range(n-3):
        if li[i][j] == '2' and li[i][j+1] == '0' and li[i][j+2] == '2' and li[i][j+3] == '0':
            num += 1
for a in range(m-3):
    for b in range(n):
        if li[a][b] == '2' and li[a+1][b] == '0' and li[a+2][b] == '2' and li[a+3][b] == '0':
            num += 1
for g in range(m-3):
    for k in range(n-3):
        if li[g][k] == '2' and li[g+1][k+1] == '0' and li[g+2][k+2] == '2' and li[g+3][k+3] == '0':
            num += 1
print(num)