openpyxl模块和csv模块

openpyxl模块和csv模块

前言

   csv文件是以逗号为分隔符的纯文本文件，而 Excel 文件是 Microsoft Office 软件生成的一种电子表格文件。csv格式存储数据，读写比较方便，易于实现，文件也会比Excel文件小。但CSV 文件仅提供简单的数据存储功能，Excel 文件具有丰富的分析和计算功能。openpyxl模块是用于xlsx文件的数据操作，而csv模块是用于csv文件。

---

一、openpyxl模块

1.简介

   openpyxl 是一个用于处理 xlsx 格式 Excel 表格文件的第三方 python 库，其支持 Excel 表格绝大多数基本操作。安装，使用pip install openpyxl 命令。

2.方法

（1）文件写入方法，Workbook()

from openpyxl import Workbook,load_workbook

# Workbook()函数创建新的workbook（工作簿）对象，就是创建新的空的Excel文件。
wb = Workbook()
# wb.active就是获取这个工作簿的活动表，通常就是第一个工作表。
#创建完新的工作簿后，还得获取工作表。不然程序会无所适从，不知道要把内容写入哪张工作表里
ws = wb.active
#使用title给工作表重命名
ws.title = 'newSheet' 

# 添加完工作表，可以操作单元格，往单元格添加东西
ws['A1'] = '神话故事'
# 把我们想有写入的内容写成列表，赋值给row
row = ['盘古',"女娲","夸父"]
# 使用ws.append（）往表格里添加这一行文字，append为添加，列表一个元素为一个单元格
ws.append(row)

# 先把要写入的多行内容写成列表，再放大进入大列表里，赋值给rows
rows = [['盘古',"女娲","夸父"],['是','神话','故事', '经典','人物']]
for i in rows:
    ws.append(i)
print(rows)

a = [1,2,3,4]
b = ["A","B","C","D"]
c = ["!","@","#","$"]
#添加列名
ws.append(['a','b','c'])
#使用为zip函数把多个列表的值进行迭代合并
for i in list(zip(a,b,c)):
    ws.append(i)

# 保存新建的Excel文件，并命名为“Myth.xlsx”
wb.save('C:/Users/tang/Desktop/Myth.xlsx')

（2）文件读取方法，load_workbook()

#使用load_workbook（）函数，该方法中还有一个 read_only 参数用于设置文件打开方式。
#默认为可读可写，该方法最终将返回一个 workbook 的数据对象
wb = load_workbook('C:/Users/tang/Desktop/Myth.xlsx')
ws = wb['newSheet'] 
#sheetnames是用来获取工作簿所有工作表的名字的。
#如果你不知道工作簿到底有几个工作表，就可以把工作表的名字都打印出来。
sheetname = wb1.sheetnames
print(sheetname)

#1. 获取A1单元格的值
A1_cell = ws['A1']
A1 = A1_cell.value
# 2.通过sheet.cell(row,column)获取,即sheet（'A2'）
A2_cell = ws.cell(2,1)
A2 = A2_cell.value
# 3.获取单元格所在行和列
print((A2_cell.row,A2_cell.column))
#4.对行进行遍历
for row in ws.rows:
    for cell in row:
        print(cell.value)
# 5.对列进行遍历
for column in ws.columns:
    for cell in column:
        print(cell.value)
# 6.对特定范围进行遍历,如获取第一行的值
for cell in list(ws.rows)[0]:
    print(cell.value)
# 7.对某一单元格的范围进行遍历：
for spaces in sheet['A1':'B2']:  #获取A1，A2，B1,B2的值
    for cell in spaces:
        print(cell.value)
# 8.通过max_row 和 max_column 来获取获取表格行和列的数量
print(ws.max_row)
print(ws.max_column)

#在读取文件时需要加上参数 data_only=True ，这样才能返回数字，
#否则将返回字符串，即公式本身
# 直接赋值
sheet['A1'].value = 2
# 公式赋值
sheet['A6'].value = '=SUM(A1:A5)'

（3）字体、边框、颜色的调整

# 导入字体、边框、颜色以及对齐方式相关库
from openpyxl.styles import Font, Border, Side, PatternFill, colors, Alignment

# 1.通过 sheet 单元格 font 属性设置字体风格
# 设置字体风格为Times New Roman，大小为16，粗体、斜体，颜色蓝色
sheet['A1'].font = Font(name='Times New Roman', size=16, bold=True, italic=True, color=colors.BLUE)
# 2.通过 sheet 单元格 alignment 属性设置文本对齐风格
# 通过参数horizontal和vertical来设置文字在单元格里的对齐方式，此外设置值还可为left和right
sheet['B1'].alignment = Alignment(horizontal='center',vertical='center')

二、csv模块

1.简介

    ​ CSV(Comma Separated Values)是逗号分隔符文本格式，常用于Excel和数据库的导入和导出，Python内置的CSV模块提供对CSV格式文件读取和写入的对象。

---

2.方法

（1）写入writer()

import csv

rows=[("铅笔","20"),("毛笔","21"),("钢笔","22")]
# 调用open()函数打开csv文件，传入参数：文件名“demo.csv”、
# 写入模式“w”、newline=''、encoding='utf-8'。
with open('demo.csv','w',newline='',encoding='utf-8') as f:
	# 1.用csv.writer()函数创建一个writer对象
	writer = csv.writer(f)
	# 2.调用writer对象的writerrow（）方法，可以在csv文件里写入一行文字
	writer.writerow(['电影',"豆瓣评分"])
	writer.writerow(['银河护卫队','8.0'])
	writer.writerow(['复仇者联盟','8.1'])
	#3.多行写入
    w.writerows(rows)
# 写入完成,关闭文件
f.close()

#只能读取原文件render，修改所对应的列的值，写入新的文件中
reader = csv.reader(in_file)
out_file = open("d:/out.csv", "wb")
writer = csv.writer(out_file)
for row in reader:
    row[3] = 4
    writer.writerow(row)
out_file.close()
#修改特点的单元格
 f = open('mylist.csv', 'r')
 reader = csv.reader(f)
 mylist = list(reader)
 f.close()
 mylist[1][3] = 'X'
 my_new_list = open('mylist.csv', 'w', newline = '')
 csv_writer = csv.writer(my_new_list)
 csv_writer.writerows(mylist)
 my_new_list.close()

（2）读取reader()

with open('demo.csv', 'r', newline='', encoding='utf-8') as f:
	# 1.用csv.reader()函数创建一个reader对象
    reader = csv.reader(f)
    #2.遍历打印每一行
    for r in reader:
        print(r)      

# 数据
No.,Name,Age,Score
1,Apple,12,98
2,Ben,13,97
3,Celia,14,96
4,Dave,15,95
with open('A.csv','rb') as csvfile:
    reader = csv.reader(csvfile)
    rows = [row for row in reader]   
print(rows) 	

[['No.', 'Name', 'Age', 'Score'],
['1', 'Apple', '12', '98'],
['2', 'Ben', '13', '97'],
['3', 'Celia', '14', '96'],
['4', 'Dave', '15', '95']]

#1.读取csv文件的某一行
    for i,rows in enumerate(reader):
        if i == 2:
            row = rows            
print (row) 

['2', 'Ben', '13’', '97']

#2.读取csv文件的某一列
	column = [row[2] for row in reader]
	print(column)

['Age', '12', '13', '14', '15']

3.csv的读写模式

（1）r:以读方式打开文件,可读取文件信息
（2）w: 已写方式打开文件,可向文件写入信息。如文件存在,则清空,再写入
（3）a：以追加模式打开文件，打开文件可指针移至末尾，文件不存在则创建
（4）r+:以读写方式打开文件，可对文件进行读和写操作
（5）w+:消除文件内容,以读写方式打开文件
（6）a+：以读写方式打开文件，文件指针移至末尾
（7）b:以二进制打开文件