i = 0
def getNum():
global i
i = i + 1
return i
def genMatrix(rows, cols):
matrix = [[getNum() for col in range(cols)] for row in range(rows)]
#for i in range(rows):
# for j in range(cols):
# print matrix[i][j],
# print '\n'
return matrix
def printMatrix(matrix):
size = len(matrix)
length = 2 * size - 1
for k in range(0, length):
for i in range(0, size):
for j in range(0, size):
if j-i == size - 1 - k:
print matrix[i][j],
print ''
if __name__ == '__main__':
matrix = genMatrix(4, 4)
printMatrix(matrix)