[Numpy-discussion] Retrieving and flattening lower triangular components of inner axes of 3d array

Damien Moore damienlmoore at gmail.com
Fri May 25 10:19:34 EDT 2012


Hi List,

I have an array, x, with dimensions (K,M,M). I would like to compute a
new array, y ~ (K,M*(M+1)/2), such that

y[k] = numpy.tril(x[k]).flatten() for each k = 0,..,K-1

Is there a way to do this without looping? I was trying with
tril_indices but couldn't quite get the logic right.

Thanks,
dm



More information about the NumPy-Discussion mailing list