Вернуть DataFrame из функции в другой файл

Question

Я пытаюсь создать два файла: один, который создаст серию фреймов данных, и другой, который импортирует эти фреймы данных в мой основной файл.

Примерно так:

load_data.py

def data_mean():
    import pandas as pd

    global mean_re5200, mean_re2000

    mean_re5200=pd.read_csv('mean_re5200.csv')
    mean_re2000=pd.read_csv('mean_re5200.csv')

main_project.py

from load_data import data_mean

Когда я запускаю файл main_project и набираю data_mean () в терминале, все выглядит нормально, но кадры данных не сохраняются в качестве локальных переменных, которые я могу использовать их. Я видел другие подобные цитаты здесь, в StackOverFlow, но никто не говорил о сохранении данных, только простые переменные.

Как я могу продолжить?

Answer 1

Почему бы вам просто не попробовать что-то вроде

load_data.py

import pandas as pd

df = pd.DataFrame({"a":list(range(10))})

main.py

from load_data import *

print(df)

или альтернативно

load_data.py

import pandas as pd

def data_mean():
    df0 = pd.DataFrame({"a":list(range(10))})
    df1 = pd.DataFrame({"b":list(range(10))})
    return df0, df1

main.py

from load_data import data_mean

df1, df2 = data_mean()
print(df1)