Python数据持久化-csv、excel篇

2018年7月4日笔记 学习目标: 1.会使用Python第三方模块操作CSV文件 2.会使用Python第三方模块操作EXCEL文件

本章内容: Python操作CSV:什么是CSV、Python如何操作CSV文件、Python如何写入CSV文件 Python操作EXCEL:利用xlrd模块操作Excel、利用xlwt模块写入EXCEL、xlutils结合xlrd操作EXCEL

Python操作CSV

1.什么是CSV

CSV是一种通用的,相对简单的文件格式,被用户、商业和科学广泛应用。最广泛的应用是在程序之间转移表格数据,而这些程序本身是不兼容的格式上进行操作的。因为大量程序都支持CSV变体,至少是作为一个可选择的输入/输出格式。 例如,一个用户可以需要交换信息,从一个私有格式存储数据的数据库程序,得到一个数据格式完全不同的电子表格。最有可能的情况是,该数据库程序被导出数据为“CSV”,然后被导出的CSV文件可以被电子表格程序导入。 “CSV”并不是一种单一的、定义明确的格式。因此在实践中,术语“CSV”泛指具有以下特征的任何文件:

  1. 纯文本,使用某个字符,比如ASCII、Unicode、EBCDIC或GB2312,由记录组成;
  2. 每条记录被分隔符分隔为字段(典型分隔符有逗号、分号或制表符);
  3. 有时分隔符可以包括可选的空格;
  4. 每条记录都有同样的字段序列。

2.读取csv文件

运行代码的py文件所在的文件夹需要有销售相关企业信息.csv这个文件, 此文件下载链接: https://pan.baidu.com/s/1j8_kSiEFoB_lrLPNSJAiSw 密码: 8248 文件下载后要和被执行的py文件放到同一级目录

import csv

if __name__ == "__main__":
    with open("销售相关企业信息.csv") as csvFile:
        reader = csv.reader(csvFile)
        print(type(reader))
        for i in reader:
            print(i,type(i))

上面一段代码的运行结果如下:

<class '_csv.reader'>
['类别编号', '类别名称', '说明'] <class 'list'>
['1', '饮料', '软性饮料,咖啡,啤酒,及麦酒'] <class 'list'>
['2', '调味品', '甜酸酱,配料,涂料,及香料'] <class 'list'>
['3', '点心', '甜点心,糖果,甜面包'] <class 'list'>
['4', '日用品', '寿司'] <class 'list'>
['5', '谷类/麦片', '面包,饼干,面团,麦片'] <class 'list'>
['6', '肉/家禽', '肉品'] <class 'list'>
['7', '特制品', '水果干及豆腐'] <class 'list'>
['8', '海鲜', '海带及鱼类'] <class 'list'>

3.写入csv文件

import csv

if __name__ == "__main__":
    with open("./csvTest.csv","w") as csvfile:
        writer = csv.writer(csvfile)
        writer.writerow(['id','url','keywords'])
        data = [
            ['1','http://www.baidu.com','百度'],
            ['2','http://www.tmall.com','天猫'],
            ['3','http://www.jd.com','京东']
        ]
        writer.writerows(data)

上面一段代码运行产生的csv文件如下图所示:

图片.png-7.6kB

让产生的csv文件不空行,只需要在open函数内添加参数newline=''

import csv

if __name__ == "__main__":
    with open("./csvTest.csv",\
              "w",newline='') as csvfile:
        writer = csv.writer(csvfile)
        writer.writerow(['id','url','keywords'])
        data = [
            ['1','http://www.baidu.com','百度'],
            ['2','http://www.tmall.com','天猫'],
            ['3','http://www.jd.com','京东']
        ]
        writer.writerows(data)

上面一段代码运行产生的csv文件如下图所示:

图片.png-4.4kB

4.小结

简述Python如何读写csv open()函数产生的文件赋值给file变量. 用csv.reader(file)产生一个reader用来读文件,reader是一个可迭代对象,数据类型为csv.reader 用csv.writer(file)产生一个writer用来写文件,reader是一个可迭代对象,数据类型为csv.writer

Python操作Excel

1.利用xlrd模块读取Excel

运行代码的py文件所在的文件夹需要有某公司贸易数据.xlsx这个文件, 此文件下载链接: https://pan.baidu.com/s/1shU7evZJ8KFEmNAqoMwlBw 密码: wr7q 文件下载后要和被执行的py文件放到同一级目录

1.1 读取excel文件工作表中第1个工作薄的第1行。

import xlrd

if __name__ == "__main__":
    excel = xlrd.open_workbook('某公司贸易数据.xlsx')
    sheet = excel.sheets()[1]
    row = sheet.row_values(0)
    print(row)

上面一段代码的运行结果如下:

['产品编号', '产品', '供应商编号', '类别编号', '单位数量', '单价', '库存量', '已订购量', '安全存量', '不再销售']

1.2 读取excel文件工作表中第2个工作簿的第4行第3列的数据

import xlrd

if __name__ == "__main__":
    excel = xlrd.open_workbook('某公司贸易数据.xlsx')
    sheet = excel.sheets()[1]
    cell = sheet.cell(3,2)
    print(cell,type(cell))
    cellValue = sheet.cell(3,2).value
    print(cellValue,type(cellValue))

上面一段代码的运行结果如下:

text:'1' <class 'xlrd.sheet.Cell'>
1 <class 'str'>

1.3 对excel文件的每一行进行循环遍历

import xlrd

if __name__ == '__main__':
    excel = xlrd.open_workbook("某公司贸易数据.xlsx")
    sheet = excel.sheet_by_index(0)
    for i in range(sheet.nrows):
        print(sheet.row_values(i))

上面一段代码的运行结果如下:

['类别编号', '类别名称', '说明'] ['1', '饮料', '软性饮料,咖啡,啤酒,及麦酒'] ['2', '调味品', '甜酸酱,配料,涂料,及香料'] ['3', '点心', '甜点心,糖果,甜面包'] ['4', '日用品', '寿司'] ['5', '谷类/麦片', '面包,饼干,面团,麦片'] ['6', '肉/家禽', '肉品'] ['7', '特制品', '水果干及豆腐'] ['8', '海鲜', '海带及鱼类']

1.4 对excel文件的每一列进行循环遍历

import xlrd

if __name__ == '__main__':
    excel = xlrd.open_workbook("某公司贸易数据.xlsx")
    sheet = excel.sheet_by_index(0)
    for i in range(sheet.ncols):
        print(sheet.col_values(i))

上面一段代码的运行结果如下:

['类别编号', '1', '2', '3', '4', '5', '6', '7', '8'] ['类别名称', '饮料', '调味品', '点心', '日用品', '谷类/麦片', '肉/家禽', '特制品', '海鲜'] ['说明', '软性饮料,咖啡,啤酒,及麦酒', '甜酸酱,配料,涂料,及香料', '甜点心,糖果,甜面包', '寿司', '面包,饼干,面团,麦片', '肉品', '水果干及豆腐', '海带及鱼类']

2.利用xlwt模块写入Excel

新建一个excel,第1行第2列写入内容excel test,保存文件,命名为excelTest.xls

import xlwt

if __name__ == "__main__":
    excel = xlwt.Workbook()
    sheet = excel.add_sheet("表01")
    sheet.write(0,1,'excel test')
    excel.save("excelTest.xls")

上面一段代码运行截图如下:

image_1chqns1mi15pnr4l1564qrdovg9.png-26.7kB

3.利用xlutils结合xlrd操作Excel

修改excel文件中的第2行第3列的值为,之前该格子的内容为"软性饮料,咖啡,啤酒,及麦酒"

import xlrd
from xlutils.copy import copy

if __name__ == "__main__":
    excel = xlrd.open_workbook("某公司贸易数据.xlsx")
    excelNew = copy(excel)
    sheet = excelNew.get_sheet(0)
    sheet.write(1,2,"酒")
    excelNew.save("某公司贸易数据(修改1).xls")

image_1chqof3tk4v4frq1dla4501fntm.png-23kB

4.利用openpyxl操作Excel

利用get_sheet_names获取所有工作簿的名字,openpyxl库访问第1行第3列是sheet.value(1,3)

from openpyxl import load_workbook

if __name__ == '__main__':
    excel = load_workbook("某公司贸易数据.xlsx")
    sheetNames = excel.get_sheet_names()
    print(sheetNames)
    sheet = excel.get_sheet_by_name("客户")
    for i in range(1,5):
        print(sheet.cell(row=i,column=3).value)
    sheet['C3'] = "张先生"
    print("修改excel文件后,再次打印:")
    for i in range(1,5):
        print(sheet.cell(row=i,column=3).value)

上面一段代码运行结果如下:

['产品类别', '产品资料', '订货明细', '订货主档', '员工', '商品供应', '客户', '货运公司'] 联系人 陈小姐 黄小姐 胡先生 修改excel文件后,再次打印: 联系人 陈小姐 张先生 胡先生

作业

image_1chqq144d12icgobgtm107hj7t13.png-65.4kB

import xlwt

if __name__ == '__main__':
    excel = xlwt.Workbook()
    sheet = excel.add_sheet("sheet1")
    table = [['姓名','分数'],['小明','80'],['小王','90'],['小刚','100']]
    for i in range(len(table)):
        for j in range(len(table[i])):
            sheet.write(i,j,table[i][j])
    excel.save("excelTest.xls")

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏小白课代表

学术党!EndNote 8.2 文献管理软件 中文汉化版 安装教程

16250
来自专栏吉浦迅科技

DAY48:阅读 Atomic Functions

An atomic function performs a read-modify-write atomic operation on one 32-bit o...

10110
来自专栏Python中文社区

怎样用Python实现地理编码

怎样用Python实现地理编码 专栏作者:时空Drei ❈ 时空Drei,德国德累斯顿工业大学在读博士生,个人的擅长领域为:利用Python进行空间数据(遥感G...

45270
来自专栏linux、Python学习

玩这么久 Python ,这些好玩又实用的库一定不能错过!

21700
来自专栏喵了个咪的博客空间

[喵咪软件推荐(2)]全球服务器测速工具speedtest-cli

[喵咪软件推荐(2)]全球服务器测速工具speedtest-cli w-blog博客 ? 前言 在上一篇章中给大家介绍了全球国家信息国旗区号等一些全球国家新的库...

1.1K70
来自专栏北京马哥教育

给Python新人练手准备的十个简单趣味脚本

? 01. 目录文件分类 前言 有时候,想要对一个目录里的文件进行搜索或者分类操作往往是一件痛苦的事情,下面这个程序的目的是将目录下的文件树以某种分类规则进行...

523100
来自专栏吉浦迅科技

【阿星的学习笔记(1)】如何在windows安裝Theano +Keras +Tensorflow並使用GPU加速訓練神經網路

今天开始,Lady向各位介绍一个朋友阿星(Ashing)以及他的机器学习读书笔记! ? 阿星也是我们手撕深度学习算法微信群的热心群友!接下来,Lady我也会陆续...

47960
来自专栏AI研习社

Github 项目推荐 | Twitter 开源数据分析工具 —— tinfoleak

Tinfoleak 是 OSINT (Open Source Intelligence) 和 SOCMINT (Social Media Intelligenc...

19030
来自专栏木宛城主

SharePoint 2013 Designer工作流——Parallel Block的应用

参考目录 安装和配置SharePoint 2013 Workflow SharePoint 2013 实现多级审批工作流 在自定义Workflow...

278100
来自专栏施炯的IoT开发专栏

PhoneFinder--寻找丢失的手机

    手机丢了怎么办?那就打电话给手机,如果运气好的话,捡到的好心人能够把手机还给你。如果手机是被偷的,那就没有办法了,即使手机开着,估计小偷也不会接电话。当...

29140

扫码关注云+社区

领取腾讯云代金券