How To Take Matrix Input In Python