python: setting an element of a numpy matrix,"a[i, j] = 5"