Как суммировать по двум индексам в python?

Question

Может ли кто-нибудь помочь мне с функцией sum?

Я пытаюсь суммировать по двум индексам. Я хочу, чтобы получить следующий результат:

p_t[0, 0]+p_t[0, 1]+p_t[0, 2]+p_t[1, 0]+p_t[1, 1]+p_t[1, 2]+p_t[2,0]+p_t[2, 1]+p_t[2, 2] 

, используя этот код:

num_products=3 
sum (p_t[i, j] for i,j in range(num_products)) 

Но я получаю ошибку:

TypeError Traceback (most recent call last) in() ----> 1 sum (p_t[i, j] for i,j in range(num_products)) in (failed resolving arguments) TypeError: 'int' object is not iterable

Я также попытался это, но не работал ни

sum (p_t[i, j] for i in range(num_products) for j in range(num_products)) 

Answer 1

sum(p_t[i][j] for i in range(len(p_t)) for j in range(len(p_t[i]))) 

Answer 2

import numpy as np 
from itertools import product 

a =[[1,2,3],[4,5,6],[3,3,3]] 
p_t = np.array(a) 
num_products = 3 

rslt1 = sum (p_t[i, j] for i,j in product(range(num_products), repeat=2)) 

rslt2= sum (p_t[i, j] for i in range(num_products) for j in range(num_products)) 

Выход:

rslt1 
Out[79]: 30 

rslt2 
Out[80]: 30 

Для первой попытки, вы можете использовать itertools.product.

Для вашей второй попытки, на самом деле, ваш sum (p_t[i, j] for i in range(num_products) for j in range(num_products)) работает с моей стороны.

---

Как правило, ниже два эквивалентен, когда речь идет о продукте два списка

In [88]: [(i,j) for i in range(2) for j in range(3)] 
Out[88]: [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2)] 

In [89]: [x for x in product(range(2), range(3))] 
Out[89]: [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2)]