将数据框导入另一个脚本

Question

我在 python 中有一个脚本（脚本 1），它产生dataframe这样的一个：

import pandas as pd

import numpy as np

df = pd.DataFrame(np.array([[1, 2], [4, 5]]), index=('27-04-2020','28-04-2020'), columns=('Prediction', 'Certainty'))

我想导入dataframe到（脚本A）。但是，我似乎找不到导入数据框的方法。有人可以帮我解决这个问题吗？

Answer 1

您可以简单地在 script_a.py 中执行此操作from script_1 import df但更简洁的方法是：在 script_1.pyimport pandas as pdimport numpy as npdef get_dataframe():    df = pd.DataFrame(np.array([[1, 2], [4, 5]]), index=('27-04-2020','28-04-2020'), columns=('Prediction', 'Certainty'))在 script_a.pyfrom script_1 import get_dataframedf = get_dataframe()

Answer 2

您只需要添加一个 return 语句。所以 script_1.pydef get_dataframe():     df = pd.DataFrame(np.array([[1, 2], [4, 5]]), index=('27-04-2020','28-04-2020'), columns=('Prediction', 'Certainty'))         return df其余的保持不变