返回一个数据帧熊猫和从主（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 

，能够使用视觉数据帧在主，在readVision.py做出的所有更改？我想用修改后的数据帧中的主，但我不知道如何从readVision.py返回。 请帮忙！

Answer 1

尝试把你的代码readVision.py成一个函数，然后导入main.py的那个函数iside：里面你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 

然后你就可以导入并调用vision功能。

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