Re: [問題] 有關nparray index的問題

[KSJ (阿真)]

※ 引述《Serenity9078 (Serenity9078)》之銘言：
: 想請問一下
: 現在我有一個ndarray:  a=([[1,2,3],
:                           [2,3,4]])
: 我要如何才能得到 [1  這個column的index呢?
:                   2]
: 麻煩各位高手謝謝!

>>> import numpy
>>> x=numpy.array([[1,2,3],[4,5,6]])
>>> x
array([[1, 2, 3],
       [4, 5, 6]])
>>> numpy.where(x == [[1],[4]])
(array([0, 1]), array([0, 0]))
>>> x == [[1],[4]]
array([[ True, False, False],
       [ True, False, False]], dtype=bool)
>>> numpy.all(x == [[1],[4]],axis = 0)
array([ True, False, False], dtype=bool)
>>> numpy.nonzero(numpy.all(x == [[1],[4]],axis = 0))
(array([0]),)
>>> numpy.nonzero(numpy.all(x == [[1],[4]],axis = 0))[0][0]
0

請參考

--
※ 發信站: 批踢踢實業坊(ptt.cc) 
◆ From: 140.112.63.202
推 Serenity9078:謝謝您！完全解決掉我的問題了