Skip to content

Commit 3810f9d

Browse files
authored
RFC: game_theory: Minor refactoring (#763)
1 parent 835a64b commit 3810f9d

File tree

3 files changed

+7
-7
lines changed

3 files changed

+7
-7
lines changed

quantecon/game_theory/game_converters.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def _parse(string):
139139
players = [
140140
Player(
141141
payoffs2d[i, :].reshape(nums_actions, order='F').transpose(
142-
list(range(i, N)) + list(range(i))
142+
(*range(i, N), *range(i))
143143
)
144144
) for i in range(N)
145145
]
@@ -196,7 +196,7 @@ def _dump(g):
196196
for i, player in enumerate(g.players):
197197
payoffs = np.array2string(
198198
player.payoff_array.transpose(
199-
list(range(g.N-i, g.N)) + list(range(g.N-i))
199+
(*range(g.N-i, g.N), *range(g.N-i))
200200
).ravel(order='F'))[1:-1]
201201
s += ' '.join(payoffs.split()) + ' '
202202

quantecon/game_theory/logitdyn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def __init__(self, data, beta=1.0):
4949

5050
for player in self.players:
5151
payoff_array_rotated = \
52-
player.payoff_array.transpose(list(range(1, self.N)) + [0])
52+
player.payoff_array.transpose((*range(1, self.N), 0))
5353
payoff_array_rotated -= \
5454
payoff_array_rotated.max(axis=-1)[..., np.newaxis]
5555
player.logit_choice_cdfs = \

quantecon/game_theory/normal_form_game.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -640,8 +640,9 @@ def __init__(self, data, dtype=None):
640640
)
641641
for i, payoff_array in enumerate(payoff_arrays):
642642
payoff_array[:] = \
643-
data.take(i, axis=-1).transpose(list(range(i, N)) +
644-
list(range(i)))
643+
data.take(i, axis=-1).transpose(
644+
(*range(i, N), *range(i))
645+
)
645646
self.players = tuple(
646647
Player(payoff_array) for payoff_array in payoff_arrays
647648
)
@@ -663,8 +664,7 @@ def payoff_profile_array(self):
663664
np.empty(self.players[0].payoff_array.shape + (N,), dtype=dtype)
664665
for i, player in enumerate(self.players):
665666
payoff_profile_array[..., i] = \
666-
player.payoff_array.transpose(list(range(N-i, N)) +
667-
list(range(N-i)))
667+
player.payoff_array.transpose((*range(N-i, N), *range(N-i)))
668668
return payoff_profile_array
669669

670670
def __repr__(self):

0 commit comments

Comments
 (0)