[开发|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__()
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值