Matrix In Python Using Array