查找与给定条件匹配的行的索引

Question

I have the following list ( nodes ):

nodeID, x, y, z=row

I want to find the index of the row which row[0]==nodeAID .

My code is:

nindF=[line[0].index(nodeAID) for line in nodes]

but it gives me the error: TypeError: expected a character buffer object

Answer 1

nindF = [index for index, line in enumerate(nodes) if line[0].find(nodeAID) >= 0]

This will return a list of indexes of all the lines that start with nodeAID. If you only care about the index of the first line that starts with nodeAID then:

nindF = [index for index, line in enumerate(nodes) if line[0].find(nodeAID) >= 0][0]