python - 如何在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])))