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)  # 单行写入