How To Accept Input For Matrix In Python