Matrix In Python Using Arrays