有没有一种方法可以创建一个函数来转换为pandas？

import pandas as pd

def to_dataframe(func):
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        if isinstance(result, pd.DataFrame):
            return result
        else:
            return pd.DataFrame(result)
    return wrapper

@to_dataframe
def generate_data():
    data = {'Name': ['Alice', 'Bob', 'Charlie'],
            'Age': [25, 30, 35],
            'City': ['New York', 'London', 'Tokyo']}
    return data

result = generate_data()
print(result)

      Name  Age      City
0    Alice   25  New York
1      Bob   30    London
2  Charlie   35     Tokyo