python - excel文件读写

 安装对应的包：pandas、xlsrd、openpyxl

 pip install -i https://pypi.tuna.tsinghua.edu.cn/simple requests pandas   

 pip install xlrd==1.2.0

 pip install openpyxl

 

 1. 读取

 

例如：原始表格如下：

方法一（较简洁，推荐使用）：

import pandas as pd  # 导入pandas包

file = './abc1.csv'
# 读取文件 1.io:excel文件路径  2.sheet_name:sheet页签名字  3.header:略过表头（默认0：第一行，None：不略过）
# 这种方法对于xlsx、xls、csv格式的文件，均可读取
data = pd.read_excel(io=file, sheet_name='Sheet1', header=0)

print(len(data))  # 数据行的行数（不包括表头）

print(data.values)  # 表中数据（不包括表头）

for val in data.values:  # 逐行打印数据
    print(val)

print(data.head())  # 打印包括表头在内的所有数据

打印结果如下：

 方法二：

import xlrd  # 导入xlrd包

# 这种方法对xlsx、xls、csv均适用
file1 = '/Users/zhangyang/PycharmProjects/untitled/abc1.xlsx'
workbook = xlrd.open_workbook(filename=file1)  # 打开工作表
booksheet = workbook.sheet_by_index(0)  # 获取sheet页（通过索引，0：第一个）
booksheet1 = workbook.sheet_by_name('Sheet1')  # 获取sheet页（通过sheet页名称）

print(booksheet1.nrows)  # nrows获取总行数(包括表头)，nclos获取总列数
for i in range(booksheet1.nrows):
    print(booksheet1.row_values(i))  # 打印整行

for i in range(booksheet1.nrows):
    print(booksheet1.row_values(i, 1, 2))  # 打印每行中的某几列，（行号，开始列id，结束列id）

print(booksheet1.cell_value(1, 2))  # 打印某个单元格

打印结果如下：

 

 

2. 写入

 

方法一（仅支持xls、xlsx，不支持csv）：

import numpy as np
import pandas as pd

data1 = np.random.randint(0, 101, size=[5, 3])  # 数据：1-888之间，5行3列的列表
colum = ['语文', '数学', '英语']   # 列标题
index = ['a', 'b', 'c', 'd', 'e']  # 行标题

df1 = pd.DataFrame(data=data1, columns=colum, index=index)  # 创建pandas数据框架，传入数据、列标题、行标题

df1.to_excel('./salary.xls',   # excel文件名
             sheet_name='salary11',  # sheet页名字
             header=True,  # 是否展示列标题
             index=True)  # 是否展示行标题

结果如下:

 ps1：如果不展示行列标题：

import numpy as np
import pandas as pd

data1 = np.random.randint(0, 101, size=[5, 3])  # 数据：1-888之间，5行3列的列表
colum = ['语文', '数学', '英语']   # 列标题
index = ['a', 'b', 'c', 'd', 'e']  # 行标题

df1 = pd.DataFrame(data=data1, columns=colum, index=index)  # 创建pandas数据框架，传入数据、列标题、行标题

df1.to_excel('./score.xls',   # excel文件名
             sheet_name='score1',  # sheet页名字
             header=False,  # 是否展示列标题
             index=False)  # 是否展示行标题

结果如下：

 ps2：如果未传入行列标题，但又想展示，会使用默认值0123：

import numpy as np
import pandas as pd

data1 = np.random.randint(0, 101, size=[5, 3])  # 数据：1-888之间，5行3列的列表
colum = ['语文', '数学', '英语']   # 列标题
index = ['a', 'b', 'c', 'd', 'e']  # 行标题

df1 = pd.DataFrame(data=data1)  # 创建pandas数据框架，传入数据、列标题、行标题

df1.to_excel('./score.xls',   # excel文件名
             sheet_name='score1',  # sheet页名字
             header=True,  # 是否展示列标题
             index=True)  # 是否展示行标题

结果如下：

 ps3：如果一次写入多个sheet页

import pandas as pd

data1 = np.random.randint(0, 101, size=[5, 3])  # 数据1
colum1 = ['语文', '数学', '英语']   # 列标题1
data2 = np.random.randint(0, 101, size=[8, 2])  # 数据2
colum2 = ['化学', '物理']   # 列标题2

df1 = pd.DataFrame(data=data1, columns=colum1)
df2 = pd.DataFrame(data=data2, columns=colum2)


with pd.ExcelWriter('./score.xls') as writer:
    df1.to_excel(writer, sheet_name='sheet101', header=True, index=False),  # 写入sheet1
    df2.to_excel(writer, sheet_name='sheet102', header=True, index=False)   # 写入sheet2

结果如下：

   

 

方法二：

代码如下（对应读取中的方法二）：

import xlwt  # 导入包

file1 = '/Users/zhangyang/PycharmProjects/untitled/abc2.csv'

header = ['id', 'name']

content = [['1', 'zhangsan'],
           ['2', 'lisi'],
           ['3', 'wangmazi']]

# 该方法对xlsx、xls、csv均适用
workbook = xlwt.Workbook(encoding='utf-8')  # 1. 创建一个workbook对象（工作表）
booksheet = workbook.add_sheet('sheet1', cell_overwrite_ok=True)  # 2. 添加sheet页

row = 0
for col in range(len(header)):
    booksheet.write(row, col, header[col])  # 3. 写单元格（列头）

for line in content:
    row += 1
    for col in range(len(line)):
        booksheet.write(row, col, line[col])  # 3. 写单元格（内容）

workbook.save(file1)  # 4. 保存文件

写入结果如下：