How To Take Matrix Input From User In Python