查找列表的子列表中是否存在某项之后，请查找同一子列表中是否存在另一项

Question

I have several lists that contain multiple sublists that are structured like this:

shape_01 = [['Circle', 'Top'], ['Dot', 'Top']]

I have created a function that checks to see if a certain shape item is present and draws that shape.

def draw_shape(set):
     if any(('Dot') in i for i in set):
     goto(0,0)  
     dot(25)

This draws a dot at 0,0 when the function is entered like this:

draw_shape(shape_01)

I want to edit the function so it checks to see if 'Top' is present in the same sublist as the shape and then changes from drawing the dot at (0,0) to (0,100).

To sum up in the best terms I can state, how can I check if both 'Dot' and 'Top' are in the same sublist and have the code change accordingly?

Answer 1

You could do it like this. But it should be said that this approach doesn't really scale up if you want to add more attributes, shapes etc.

def goto(*pos): print(pos)
def dot(*params): print(params)

def draw_shape(lst):
    for l in lst:
        if 'Dot' in l:
            if 'Top' in l:
                goto(100, 0)
            else:
                goto(0,0)
            dot(25)

---

def draw_shape(lst):
    for l in lst:
        if 'Dot' in l and 'Top' in l:
            goto(100, 0)
            dot(25)
        elif 'Dot' in l:
            goto(0,0)
            dot(25)

Answer 2

To answer your specific question of how to check if the sublist contains multiple values, i recommend using a function, like this:

def any_sublist_contains_all(lists, values_to_find):
    return any(all(value in l for value in values_to_find) for l in lists)

It returns True if ALL values in values_to_find are present IN ANY sub_list of lists

so you can do:

if any_sublist_contains_all(set, ('Dot','Top')):
    # dot what you want

Answer 3

您可以利用python中的集合。

set(sublist).issuperset(['Top', 'Dot'])