Search list in list written in Python

Question

I have modified a simple search list in list (codes borrowed from another site) but unable to capture all the relevant information.

INPUT FILE :

data = [['a','b'], ['a','c'], ['b','d'],['a','b']]

I am searching the whole list with a 'b' in it but my algorithm only captures anything ending with a 'b'. I could not capture all 'b' whether it starts in Index(0) or Index(1).

data = [['a','b'], ['a','c'], ['b','d'],['a','b']]
search = 'b'
for sublist in data:
    if sublist[1] == search:
        print("Found it!" + str(sublist))

Below is the OUTPUT but it is missing ['b','d']. Could someone help please?

Found it!['a', 'b']
Found it!['a', 'b']

You are checking sublist[1] == search, i.e. if and only if second element of sublist is equal to search. Change it to search in sublist and it should be fine. — Chris
– Chris, Commented Sep 17, 2019 at 5:26

han solo · Accepted Answer · 2019-09-20 14:58:04Z

8

Just use in for membership check like,

>>> data = [['a','b'], ['a','c'], ['b','d'],['a','b']]
>>> for sub in data:
...   if 'b' in sub:
...     print(sub)
... 
['a', 'b']
['b', 'd']
['a', 'b']

edited Sep 20, 2019 at 14:58

answered Sep 17, 2019 at 5:27

han solo

6,6401 gold badge20 silver badges22 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

user11589714 Over a year ago

Thank you. It is working. May the force be with you.

Dipen Dadhaniya · Accepted Answer · 2019-09-17 05:27:25Z

1

Try:

data = [['a','b'], ['a','c'], ['b','d'],['a','b']]
search = 'b'
for sublist in data:
    # Edited this part
    if search in sublist:
        print("Found it!" + str(sublist))

Output:

Found it!['a', 'b']
Found it!['b', 'd']
Found it!['a', 'b']

answered Sep 17, 2019 at 5:27

Dipen Dadhaniya

4,6602 gold badges19 silver badges26 bronze badges

9 Comments

user11589714 Over a year ago

Thank you. It is working. It is very clean and clear and well structure.

user11589714 Over a year ago

Is there any difference if my input data are surrounded by parentheses?

user11589714 Over a year ago

data = [('a','b'), ('a','c'), ('b','d'),('a','b')]

Dipen Dadhaniya Over a year ago

No, it will be same, except in the output you will see "Found it!('a', 'b')" instead of "Found it!['a', 'b']".

user11589714 Over a year ago

What about a tuple? I am always confused with [], {} or () or SET too.

|

Collectives™ on Stack Overflow

Search list in list written in Python

2 Answers 2

1 Comment

9 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

9 Comments

Your Answer

Sign up or log in

Post as a guest

Related