I want to compute an expression consisting of numpy arrays, but only at a given index. For example, say