I wrote something similar a couple of years ago - https://github.com/adamisntdead/QuSimPy
Happy to answer any questions people have, including on other simulation methods other than state vector!
Happy to answer any questions people have, including on other simulation methods other than state vector!