[Numpy-discussion] Extracting column vectors from matrix.

Andrew Nesbit alnesbit at students.cs.mu.OZ.AU
Thu Sep 4 02:59:05 EDT 2003


Say that I have a rank-2 array (matrix) of shape (m,n).  Call it A.

When extracting column vectors from A, I often want to keep the result
as a rank-2 column vector (for subsequent matrix multiplications,
etc), so I usually end up writing something ugly like this:

column_vector = reshape(A[:,col],(m,1))

I've got a function to wrap this, but is there a builtin (or cleaner)
way to do this sort of operation?

TIA.

Andrew.





More information about the NumPy-Discussion mailing list