[开发|python] pandas操作csv示例

1. 示例代码

#!/usr/bin/python
# -*- coding: UTF-8 -*-

"""
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 csv.reader(open(fileName,'r'))
    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__()