普通幻方（循环）

Question

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

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

以下是我的行总和方法：

^{pr2}$

我想知道怎样求两条对角线的和。在

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