locate items in matrix (index of lists of lists)

MRAB google at mrabarnett.plus.com
Fri Mar 20 10:28:07 EDT 2009


Chris Rebert wrote:
> On Fri, Mar 20, 2009 at 4:34 AM, Alessandro Zivelonghi
> <zasaconsulting at gmail.com> wrote:
>> Many Thanks guys!
>>
>> and what if I need to look ONLY into the second and third columns,
>> excluding the first item of each rows?
>>
>> for example if x = 3 I need to get  [0] and not [0,1]
> 
> indices = [i for i, row in enumerate(mat) if 1<= i <= 2 and 3 in row[1:]]
> 
If he wants to look in only the second and third columns, but still all
the rows, surely that's:

     indices = [i for i, row in enumerate(mat) if x in row[1 : 3]]



More information about the Python-list mailing list