Pythonの行列でi番目の行とi番目の列の合計が同じであるかどうかを確認します

2D行列があるとします。 i番目の行の合計がi番目の列の合計と同じであるかどうかを確認する必要があります。

したがって、入力が次のような場合

その場合、最初の行と列の合計は（2 + 3 + 4 + 5）=14および（2 + 10 + 1 + 1）=14であるため、出力はTrueになります。

これを解決するには、次の手順に従います-

• 行：=マットの行数
• col：=マットの列数
• total_row：=0、total_col：=0
• 0から行-1の範囲のiの場合、実行
  • total_row：=0、total_col：=0
  • 0からcol-1の範囲のjの場合、do
    • total_row：=total_row + mat [i、j]
    • total_col：=total_col + mat [j、i]
  • total_rowがtotal_colと同じ場合、
    • Trueを返す
• Falseを返す

理解を深めるために、次の実装を見てみましょう-

サンプルコード

def solve(mat):
   row = len(mat)
   col = len(mat[0])
   total_row = 0
   total_col = 0
   for i in range(row):
      total_row = 0
      total_col = 0
      for j in range(col):
         total_row += mat[i][j]
         total_col += mat[j][i]
       
      if total_row == total_col:
         return True
     
   return False
 
matrix = [
   [2,3,4,5],
   [10,6,4,2],
   [1,4,6,7],
    [1,5,6,7]
]
     
print(solve(matrix))

入力

[    
[1,2,3,4],     
[9,5,3,1],     
[0,3,5,6],    
[0,4,5,6]
]

出力

True