Попытка получить диагональную сумму

Question

Попытка получить сумму по диагонали в # 7 списков, созданных # 6. Так говорят, что список был этот [[2,3,1],[1,1,1],[5,6,4]] сумма будет 2 + 1 + 4

#6 

def randomlists(s): 
    b=s 
    list1 = [] 
    while s>0: 
     sublist = [] 
     for x in range(0,b): 
      sublist.append(randrange(-100,101)) 
     list1.append(sublist) 
     s-=1 
    return list1 
#print(randomlists(5)) 
#7 

def diagonalsum(x): 
    a=randomlists(x) 
    count=0 
    print (a) 
    for b in a: 
     while count<x: 
      d=a[0] 
      b=a[1] 
      c=a[2] 
      print (a[(count)]) 
      count+=1 
     return d+b+c 
print (diagonalsum(3)) 

Answer 1

Чтобы получить диагональные элементы вы берете первый в первом ряду, второй во втором ряду ... п-й в n-я строка.

def diagonalsum(x): 
    a = randomlists(x) 
    return sum(row[i] for i, row in enumerate(a)) 

Не настолько кратким альтернатива, которая использует нормальный цикл читает

def diagonalsum(x): 
    a = randomlists(x) 
    result=0 
    for i, row in enumerate(a): 
     result += row[i] 
    return result 

Answer 2

Предполагая, что матрица является квадратной, это стандартное решение с использованием цикла:

def diagonalsum(m): 
    count = 0 
    for i in xrange(0, len(m)): 
    count += m[i][i] 
    return count 

... Который может быть написан более сжатым способом, используя выражения генератора и функцию sum:

def diagonalsum(m): 
    return sum(m[i][i] for i in xrange(len(m))) 

Answer 3

Использование numpy.trace: http://docs.scipy.org/doc/numpy-1.10.1/reference/generated/numpy.trace.html. Он возвращает сумму по диагоналям массива.

import numpy as np 

M = np.array([[2,3,1],[1,1,1],[5,6,4]]) 

print(M.trace()) 

#7