range append
import numpy as np
rc=list(map(int,input().split()))
r=rc[0]
c=rc[1]
l=[]
for i in range(r):
l.append(list(map(int,input().split())))
k=np.zeros((r,c),int)
D=[]
for i in range(r):
for j in range(c):
x=i
y=j
if k[i][j]==0:
d=[]
while x<r and y<c:
d.append(l[x][y])
k[x][y]=1
x+=1
y+=1
D.append(d)
for i in range(r):
print(l[i])
print("\nDiagonals are:")
for i in range(r+c-1):
print(D[i])
max=0
for i in D:
if sum(i)>max:
max=sum(i)
ind=D.index(i)
print(ind,D[ind],max)
Parsed in 0.005 seconds