如何檢查是否在對角矩陣等於

Question

我想打一個井字遊戲。

我在對角的麻煩。

我已經成功找到一種方法來檢查，如果列或行是用一個for循環相等。

我需要一種方法來檢查，如果一個對角線等於每一個來看看是否有人贏了的舉動。

Answer 1

惠康計算器！

我建議使用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)