返回一個數據幀熊貓和從主（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