Python csv文件操作

一、open文件打开和with open as 文件打开的区别

file= open("test.txt","r")
try:
    for line in file.readlines():
        print(line)
except:
    print("error")
finally:
    file.close()

with open("test.txt","r") as file:
    for line in file.readlines():
        print(line)

用with语句的好处，就是到达语句末尾时，会自动关闭文件，即便出现异常。

with语句实际上是一个非常通用的结构，允许你使用所谓的上下文管理器。上下文管理器是支持两个方法的对象：__enter__和__exit__。

方法__enter__不接受任何参数，在进入with语句时被调用，其返回值被赋给关键字as后面的变量。

方法__exit__接受三个参数：异常类型、异常对象和异常跟踪。它在离开方法时被调用（通过前述参数将引发的异常提供给它）。如果__exit__返回False，将抑制所有的异常。

文件也可用作上下文管理器。它们的方法__enter__返回文件对象本身，而方法__exit__关闭文件

with语句作用效果相当于上面的try-except-finally

二、csv文件的读取

with open('test.csv', 'r') as csv_file:
    reader = csv.reader(csv_file)
    #print(list(reader))
    for item in reader: #按行读取
        print (item)

>>> list(reader)---> [['name','age'], ['Bob','14'], ['Tom','23'], ...]
>>> item -->
    ['name','age']
    ['Bob','14']
    ['Tom','23']
    ...    

　　

三、csv文件的写入

import csv
header = ['name','age']
data = [
    ['Bob',14],
    ['Tom',23],
    ['Jerry',18]
]
with open('test.csv', 'w', newline='')as csv_file:
    csv_writer = csv.writer(csv_file)
    csv_writer.writerow(header)
    # csv_writer.writerows(data) # 写入多行
    for row in data:
        csv_writer.writerow(row) # 单行写入

指定newline='',则可以避免每写入一行就写入一空行。

四、DictReader和DictWriter操作csv

使用DictReader可以像操作字典那样获取数据，把表的第一行（一般是标头）作为key。可访问每一行中那个某个key对应的数据。

with open('test.csv', 'r')as csv_file:
    dict_reader = csv.DictReader(csv_file)
    name_list = [row['name'] for row in dict_reader]
    print(name_list)
    print(len(list(dict_reader)))

>>> ['Bob', 'Tom', 'Jerry']
>>> 0
dict_reader只能迭代一次

header = ['name', 'age']
data = [
    {'name': 'Bob', 'age': 14},
    {'name': 'Tom', 'age': 23},
    {'name': 'Jerry', 'age': 18}
]
with open('test.csv', 'w', newline='')as csv_file:
    dict_writer = csv.DictWriter(csv_file, header)
    dict_writer.writeheader()
    dict_writer.writerows(data)  # 写入多行
    for row in data:
        dict_writer.writerow(row)  # 单行写入