- 示例代码
"""
pip install pandas
"""
import pandas as pd
carBalanceFile = "carBalance.csv"
carUserInfoFile = "carUserInfo.csv"
teacherInfoFile = "teacherInfo.csv"
studentInfoFile = "studentInfo.csv"
def getCsv(fileName ,col =None):
return pd.read_csv(fileName , usecols = col)
def getCarBalance(file):
return getCsv(file , col = [1,4,0,2,7,8,9,10])
def getCarUserInfo(file):
return getCsv(file , col = [0, 1, 2, 9, 14, 15])
def getTeacherInfo(file):
return getCsv(file, col = [0, 1, 2])
def getStudentInfo(file):
return getCsv(file,col = [0, 1, 2, 21])
def gatherData():
balance_df = getCarBalance(carBalanceFile)
car_user_df=getCarUserInfo(carUserInfoFile)
teacherInfo_df = getTeacherInfo(teacherInfoFile)
studentInfo_df = getStudentInfo(studentInfoFile)
car_Info_df = pd.merge(balance_df,car_user_df,how='inner')
student_car_info=pd.merge(car_Info_df,studentInfo_df,how='inner')
student_car_info.to_csv("test1.csv", index=False, sep=',')
teacher_car_info=pd.merge(car_Info_df,teacherInfo_df,how='inner')
teacher_car_info.to_csv("test2.csv", index=False, sep=',')
gater_info=pd.merge(student_car_info,teacher_car_info,how='outer')
gater_info.to_csv("gaterInfo.csv", index=False, sep=',')
def __main__():
gatherData()
__main__()