将数据帧存储在变量 python 中

Question

我仍在学习 python，我有一个简单的问题，我已经进行了研究，但没有成功。

我有这个简单的代码：

import pandas as pd


def readDF():
  df = **reads  a df from excel**
  return df

DF = readDF()

问题是，每次我尝试从另一个模块访问 DF 变量时，它都会再次读取整个数据帧，而我只想读取一次并“存储它”以供使用。 有没有办法在第一次读取它并将其“存储”在变量 DF 中？

Answer 1

您可以将数据加载到一次调用函数的数据存储类中，或者根本不使用函数：

import pandas as pd

# Option 1: call function
class DataStorage:
    def __init__(self):

    self.df = self.readDF()        

    def readDF(self):
         df = pd.read_excel('test.xlsx')
        return df

# Option 2: as attribute:
class DataStorage:
    def __init__(self):
        
        self.df = pd.read_excel('test.xlsx')

要访问任一选项中的数据，您将在调用模块中：

from data_module import DataStorage

data = DataStorage()
data.df  # your dataframe