Normal Magic Square（By Loop）

Question

有谁知道如何实现一个函数来确定一个矩阵是否是一个魔方？例如：

>>> is_magic_square([[2, 7, 6], [9, 5, 1], [4, 3, 8]]) 
True 

这里是我行的总和方法：

def row_sum(rows): 
    return (list (map (lambda x: sum(x), rows))) 

我想知道如何得到两条对角线的总和。

Answer 1

rows = all(sum(row) == sum(matrix[0]) for row in matrix) 
cols = all(sum(col) == sum(map(list, zip(*matrix))[0]) for col in matrix) 

diag1 = sum(matrix[i][i] for i in range(3)) == sum(matrix[0]) 
diag1 = sum(matrix[2-i][i] for i in range(3)) == sum(matrix[0]) 

print rows and cols and diag1 and diag2 # magic or not