如何检查矩阵中的对角线是否相等

时间:2017-03-08 11:43:03

标签: python

我想做一个tic tac toe游戏。

我遇到对角线问题。

我设法找到一种方法来检查列或行是否与for循环相等。

我需要一种方法来检查一个对角线是否相等,以确定是否有人赢了。

1 个答案:

答案 0 :(得分:0)

欢迎使用StackOverFlow!

我建议使用numpy包。

x = np.arange(9).reshape((3,3))
x =
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
di = np.diag(x)

di
array([0, 4, 8])

现在我们需要check函数列表:

def checkEqual(lst):
   return lst[1:] == lst[:-1]

在你可以查看列表是否“相等”之后:

checkEqual(di)