Значения значений словаря

Question

Я пытаюсь суммировать значения для фрагмента словаря со значениями класса int.

Вот мой код:

from scipy.stats import poisson 

inventories = 4 # states 
mu = 1 # value for lambda 
prob = [] 
for i in range(inventories): 
    prob.append(poisson.pmf(i, mu)) 

Transitions = {} 
for i in range(inventories): 
    for j in range(inventories-1,-1,-1): 
     if i - j < 0: 
      Transitions[0,i,j] = 0 
     elif j <> 0: 
      Transitions[0,i,j] = prob[i-j] 
     elif j == 0: 
      Transitions[0,i,j] = 1 

При запуске этого я пытаюсь получить последнюю строку кода просуммировать все J для каждого I. Я кодер MATLAB, поэтому я привык к матрицам, и я думаю, что это меня завораживает. Буду признателен за любую оказанную помощь.

Answer 1

Вы можете нарезать словарь, используя списковых:

>>> l=[v for (k,v) in Transitions.iteritems() if k[1] == 3] 
>>> sum(l) 
1.9196986029286058 

Это дает сумму по J для г = 3. просуммировать I, заменить k[1] с k[2].

Все сказанное, такие списки, как это, довольно быстрые, хотя я не могу сказать, как они относятся к scipy-матрицам.

Вы также можете получить фантазии:

>>> l=[sum([v for (k,v) in Transitions.iteritems() if k[1] == i]) for i in range(1,4,1)] 
>>> l 
[1.3678794411714423, 1.7357588823428847, 1.9196986029286058] 

Это дает вам сумму по каждой строке.

---

Таким образом, вышеизложенное основано на моем непонимании вопроса. К сожалению, ОП :(

Я думаю, что вы хотите заменить строку с чем-то вроде:

Transitions[0,i,j] = sum([Transitions[0,i,k] for k in range(1,i+1,1)])