Matrix Question In Python