How To Take Matrix As Input In Python