Finding index locations in a list of tuples

Question

I have a list of tuples that looks somewhat like this

global_list = [('Joe','Smith'),('Singh','Gurpreet'),('Dee','Johnson'),('Ahmad','Iqbal')..........]

I want to find index location in global_list of

Tuples that have 'John' in them
Tuples that have 'Richard' or 'Thomas' or 'Khan' in them

The tuple could be ('First Name','Last Name') or ('Last Name','First Name').

Thanks in advance

Does this answer your question? [Find an Exact Tuple Match in a List of Tuples and Return Its Index](https://stackoverflow.com/questions/20239312/find-an-exact-tuple-match-in-a-list-of-tuples-and-return-its-index) — deadvoid, Mar 01 '21 at 00:19

score 1 · Answer 1 · answered Mar 01 '21 at 00:25

As I understood you want to find indexes. In this situation you need to use enumerate.

indexes_1 = []
indexes_2 = []
for i, tup in enumerate(global_list):
    if "John" in tup:
        indexes_1.append(i)
    if "Richard" in tup or "Thomas" in tup or "Khan" in tup:
        indexes_2.append(i)

score 0 · Answer 2 · answered Mar 01 '21 at 00:19

0

You can use np.argwhere(np.array(gloabl_list) == name)[:,0]. To add in more conditions you can either do this for all the names or you can say:

global_list = np.array(gloabl_list)
np.argwhere((global_list == name1) | (global_list == name2) ...)[:,0]

answered Mar 01 '21 at 00:19

user15270287

1,123
4
11

score 0 · Answer 3 · answered Mar 01 '21 at 00:42

It seems like you might want a dictionary of names, with sets of indices for each name:

global_list = [('Joe', 'Smith'), ('Singh', 'Gurpreet'), ('Dee', 'Johnson'), ('Ahmad', 'Iqbal')]
name_dict = {}

for idx, (first, last) in enumerate(global_list):
    if first not in name_dict:
        name_dict[first] = set(idx)
    else:
        name_dict[first].add(idx)

    if last not in name_dict:
        name_dict[last] = set(idx)
    else:
        name_dict[last].add(idx)

Then, to search you could do:

names = ['Joe', 'Johnson']
indices = set()

for name in names:
    indices.update(name_dict.get(name, set()))

print(indices)
{0, 2}

print([global_list[i] for i in indices])
[('Joe', 'Smith'), ('Dee', 'Johnson')]

Finding index locations in a list of tuples

3 Answers3