如何从列表列表中检索元素及其所在的列表？

def search_element_in_lists(target_element, lists):
    result = []  # 用于存储符合要求的元素及其所在列表
    for i, lst in enumerate(lists):
        if target_element in lst:
            result.append((target_element, lst))
    return result

# 示例用法
target = 5
lists = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
search_result = search_element_in_lists(target, lists)
for element, lst in search_result:
    print(f"元素 {element} 在列表 {lst} 中。")

元素 5 在列表 [4, 5, 6] 中。