CODEWITHSUNDEEP
pythonpython-3.xlistpermutationsympy
PartialOrder
PartialOrder

Reputation: 2960

How can I apply a permutation to a list?

How one might get Sympy Permutation to act on a list? E.g.,

from sympy.combinatorics import Permutation

lst = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i']
perm = Permutation([[0, 2, 8, 6], [1, 5, 7, 3]])
# Then something like...
perm * lst   # This doesn't work. Throws AttributeError because of list

I'd like something like this that returns (in this example):

['g', 'd', 'a', 'h', 'e', 'b', 'i', 'f', 'c']

I have read https://docs.sympy.org/latest/modules/combinatorics/permutations.html, and don't see how.

Any suggestions as to how might one go about this?

Upvotes: 4

Views: 637

Answers (1)

CDJB
CDJB

Reputation: 14536

You can just do perm(lst)

>>> from sympy.combinatorics import Permutation
>>> lst = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i']
>>> perm = Permutation([[0, 2, 8, 6], [1, 5, 7, 3]])
>>> perm(lst)
['c', 'f', 'i', 'b', 'e', 'h', 'a', 'd', 'g']

Your example output seems to have the result of applying the reverse of the given Permutation to the list - if that is your required output you need to either reverse the final list or each list within the permutation.

From here:

The permutation can be ‘applied’ to any list-like object, not only Permutations.

Upvotes: 5

Related Questions