enumerate的具体用法
enumerate对象返回一个元组,第一个参数是序号,第二个是序号对应处的的enumerate对象的值
from numpy import *
>>> m=arange(9)
>>> n=m.reshape(3,3)
>>> n
array([[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])
>>> x=enumerate(n)
>>> x
<enumerate object at 0x016CA210>
>>> x.next()
(0, array([0, 1, 2]))
>>> x.next()
(1, array([3, 4, 5]))
>>>
已知矩阵N*2某一行第二列的数值a,求与a同行的第一列的数值
举例:
>>> m=mat([[1,2],[3,4],[5,6]])
>>> m
matrix([[1, 2],
[3, 4],
[5, 6]])
这里假设已知数字6,现在想知道同行的5这个数值。
>>> where(m==6)
(matrix([[2]]), matrix([[1]]))
>>> lineNumber=where(m==6)[0]
#返回元素6所在的行号
>>> lineNumber
matrix([[2]])
#取的matrix里的整数值int类型
>>> lineNumber.item()
2
>>> x=lineNumber.item()
>>> x
2
#取x行的第一列元素的值即数字5
>>> m[x,0]
5
成功取出来了。