Python(8)：文件的IO读写操作（操作普通文件/csv/excel）_csv文件操作

一、文件的IO读写操作

1.常用的文件读取标志符

举个例子：

假设现在有个1.txt的文件

我们简单的读取一下

通常操作文件，我们会使用这种方式，可以不用关闭文件，是Python的一种简洁方式

2.普通文件的读操作

关于读文件有3种方式，这里说一下区别

二、csv文件的读写操作

1.读取操作

有一个示例文件，内容如下

读取文件

import csv # 使用csv模块

file_path = 'result.csv'

# 读取csv文件
with open(file_path, 'r') as f:
    reader = csv.reader(f, delimiter=',')  # 默认的情况下, 读和写使用逗号做分隔符(delimiter)
    for row in reader:
        print(row)
1
2
3
4
5
6
7
8
9

结果

2.读取时候跳过某一行

import csv
from itertools import islice

with open('2.csv', 'r') as f:
    reader = csv.reader(f)
    for row in islice(reader, 1, None): # 跳过表头的第一行
        print(row)
1
2
3
4
5
6
7

3.列表方式写入csv文件

这个是通过列表的方式进行写入，如果有标题，需要自己一一对应

import csv

with open('some.csv', 'w') as f:
    writer = csv.writer(f)
    writer.writerow(['1', '1', '1', '1'])
    writer.writerow(['2', '2', '2', '2' ])
1
2
3
4
5
6
'
运行
运行

问题： 发现每写完一行会加入一个空行

解决办法： Python3因此需要加入在open( )中增加一个参数newline=’’

import csv

with open('some.csv', 'w',newline='') as f:
    writer = csv.writer(f)
    writer.writerow(['1', '1', '1', '1'])
    writer.writerow(['2', '2', '2', '2' ])
1
2
3
4
5
6
'
运行
运行

Python2可以把w或者a改成字节存储，可以避免这个问题，这里我封装了一下python2的写入

def write_to_csv(file_path, row, fileheaders):
    with open(file_path, 'ab+') as f:
        f.write(codecs.BOM_UTF8)
        writer = csv.writer(f)
        if not os.path.getsize(file_path):
            writer.writerow(fileheaders)
        writer.writerow(row)
    print(row)
1
2
3
4
5
6
7
8

4.字典方式写入csv文件

import csv

data = [
    {'name': '张三', 'age': 15, 'sex': '男'},
    {'name': '李四', 'age': 12, 'sex': '男'},
    {'name': '彪子', 'age': 1, 'sex': '女'},
]

keys = data[0].keys()  # 这里每行key相同，我随便取一行做表头

with open('result.csv', 'w', newline='', encoding='utf-8') as f:
    writer = csv.DictWriter(f, fieldnames=keys)

    writer.writeheader()

    # writer.writerows(data) # 按顺序写全部
    writer.writerow(data[0])  # 只写一条

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

三、excel的数据操作（非pandas模块）

原文件

1.读取excel的某个sheet的某一行的某一列

import xlrd

file = 'demo.xlsx'
sheet_name='Sheet1'

work = xlrd.open_workbook(file)
work_sheet = work.sheet_by_name(sheet_name)

print(work_sheet.cell_value(2, 0))  # 读取第3行的第1列
1
2
3
4
5
6
7
8
9

2.追加数据到新文件

# coding: utf-8 

import xlrd
from xlutils.copy import copy

file = 'demo.xlsx'
sheet_name = 'Sheet1'

work = xlrd.open_workbook(file)
work_sheet = work.sheet_by_name(sheet_name)

# 对数据表格进行复制
new_file = copy(work)
ws = new_file.get_sheet(0)

# 往新表格写数据
ws.write(4, 0, '4')
ws.write(4, 1, 'Danny')

# 保存
new_file_name = 'new_demo.xlsx'
new_file.save(new_file_name)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23