How do you extract a column from a multi-dimensional array?

前端 未结 20 2877
感情败类
感情败类 2020-12-02 03:56

Does anybody know how to extract a column from a multi-dimensional array in Python?

20条回答
  •  暖寄归人
    2020-12-02 04:37

    One more way using matrices

    >>> from numpy import matrix
    >>> a = [ [1,2,3],[4,5,6],[7,8,9] ]
    >>> matrix(a).transpose()[1].getA()[0]
    array([2, 5, 8])
    >>> matrix(a).transpose()[0].getA()[0]
    array([1, 4, 7])
    

提交回复
热议问题