Python：查找矩阵中交替元素的总和

Question

我需要找到交替位置的元素总和。 我尝试的是下面，它仅适用于给定的订单（2x3）。 另外，我如何编写一个泛化函数 SumPos(matrix,order) 从 main 中的用户输入中获取矩阵并在函数中执行操作？ 为简单起见，它可以是方阵。

numbers= [[1,4,7,6],[8,5,1,11]]

evensum = 0 #sum for even numbers

oddsum = 0 #sum for odd numbers

for i in range(len(numbers)): 

  for j in range(len(numbers[i])): #loop visit value in matrix 

    if(j%2==0): #if the number is even position

        print('even=',numbers[i][j])  
        evensum +=numbers[i][j]

    else:
      print('odd=',numbers[i][j])  
      oddsum+=numbers[i][j]
  
#print output

print('Evensum=',evensum)

print('Oddsum=',oddsum)

请帮忙

Answer 1

您可以添加一个input() ，其中用户输入离散值（例如0 or 1 ）来选择维度：

numbers= [[1,4,7,6],[8,5,1,11]]

evensum = 0
oddsum = 0
dim = int(input("Enter dimension 0/1 vertical/horizontal:\n"))

for i in range(len(numbers)):
    for j in range(len(numbers[i])):
        if dim==0:
            if(j%2==0):  
                evensum +=numbers[i][j]
            else:
              oddsum+=numbers[i][j]
        elif dim==1:
            if(i%2==0):  
                evensum +=numbers[i][j]
            else:
              oddsum+=numbers[i][j]
print(evensum, oddsum)

输出

Enter dimension 0/1 horizontal/vertical:
0
17 26