3310487509
Goto Top

Permutation matrix multiplication in Python

I'm a beginner in Python and i think i cannot seem to understand it well, so wanted to understand the concept of Permutation matrices. A and B are square and contain only a single 1 in each row. All of the rows are unique. I've added my first attempt as an answer.

def permmult(a, b):
    """Multiply two permutation matrices.  

     a,b: lists of positive integers and zero."""  
    c = 
    for row in a:
        c.append(b[-row])
    return c


I hope someone has a faster solution to this. Thanks in advance!

Content-Key: 3388232208

Url: https://administrator.de/contentid/3388232208

Printed on: April 20, 2024 at 03:04 o'clock

Member: max
Solution max Jul 26, 2022 updated at 21:29:04 (UTC)
Goto Top
Hi,

I think you should have a look at itertools: https://docs.python.org/3/library/itertools.html

import itertools
print(list(itertools.permutations([1,2,3])))

max