Вот один из способов:
actions = []
for payoff in payoffs.T:
actions.append(self.generate_action(n=n, cumulative_payoffs=cumulative_payoffs))
cumulative_payoffs += payoff
Это может быть сжато в Python 3.8 с использованием выражения присваивания, но читаемость страдает, и я лично не фанат:
actions = [
self.generate_action(
n=n,
cumulative_payoffs=(cumulative_payoffs := cumulative_payoffs + payoff)
)
for payoff in payoffs.T
]
Вероятно, есть хороший numpy способ создать массив cumulative_payoffs
без al oop, и тогда понимание списка станет проще.