将数据框导入另一个脚本

Question

我在 python 中有一个脚本（脚本 1），它产生dataframe这样的一个：import pandas as pdimport numpy as npdf = 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.py

import pandas as pd

import numpy as np

def get_dataframe():

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

在 script_a.py

from script_1 import get_dataframe

df = get_dataframe()

Answer 2

您只需要添加一个 return 语句。

所以 script_1.py

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

其余的保持不变