如何在嵌套/复杂的python中计算列表/集合中的唯一单词

def get_unique_words(data):
    unique_words = set()
    
    for item in data:
        if isinstance(item, str):
            words = item.split()
            unique_words.update(words)
        elif isinstance(item, (list, set)):
            unique_words.update(get_unique_words(item))
    
    return unique_words

data = ["hello world", ["foo bar", "hello world"], {"baz", "hello"}]
unique_words = get_unique_words(data)
print(unique_words)

{'hello', 'world', 'foo', 'bar', 'baz'}