作者KSJ (阿真)
看板Python
标题Re: [问题] 有关nparray index的问题
时间Thu May 17 12:56:14 2012
※ 引述《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
1F：推 Serenity9078:谢谢您！完全解决掉我的问题了 05/17 14:08