返回一个pandas dataframe并从main调用(Python)

Question

我有以下用于readVision.py的代码：

import pandas as pd
#Read csv files
vision = pd.read_csv('path/vision.csv')
vision = vision[vision['gaze_x'].notnull()]
vision = vision[vision['gaze_y'].notnull()]
vision = vision[vision['pupil_axis1'].notnull()]
vision = vision[vision['pupil_axis1'].notnull()]
vision = vision[vision['pupil_area'].notnull()]

#Then I make other changes to Vision
if __name__ == "__main__":
    return vision

然后，我有了main.py：

import readVision
print vision

可以在main中使用视觉数据帧，并在readVision.py中进行所有更改？我想在main中使用修改后的数据帧，但我不知道如何从readVision.py返回它。请帮帮我！

Answer 1

尝试将readVision.py中的代码放入一个函数中，然后在main.py中导入该函数：

# readVision.py
def vision(path):
  vision = pd.read_csv(path)
  vision = vision[vision['gaze_x'].notnull()]
  vision = vision[vision['gaze_y'].notnull()]
  vision = vision[vision['pupil_axis1'].notnull()]
  vision = vision[vision['pupil_axis1'].notnull()]
  vision = vision[vision['pupil_area'].notnull()]
  return vision

然后，您可以在main.py模块中导入并调用vision函数。

# main.py
from readVision import vision
v = vision("path/vision.csv")
print type(v)
pandas.core.frame.DataFrame