csv文件 读取保存操作

官方文档:https://docs.python.org/3/library/csv.html

写入
# -*- encoding: utf-8 -*-

import csv
from io import StringIO
from urllib import urlopen

# 按行元组参数写入
def writerCsv1():
    f = open("data.csv", "w")
    writer = csv.writer(f)
    for i in range(100):
        writer.writerow((i+1, i+2, i+3))
    f.close()

# 按行字典参数写入
def writerCsv2():
    f = open("data.csv", "w")
    writer = csv.DictWriter(f, ["name", "age"])
    for i in range(100):
        dct = {"name":i+1, "age": i+2}
        writer.writerow(dct)
    f.close()
    print("写入成功!")

writerCsv2()
读取

要读取的文件

# MontyPythonAlbums.csv

Name,Year
Monty Python's Flying Circus,1970
Another Monty Python Record,1971
Monty Python's Previous Record,1972
# 按行读取列表
def readerCsv1():
    # 读取网络文件
    # url = "http://www.pythonscraping.com/files/MontyPythonAlbums.csv"
    # data = urlopen(url).read()

    # 读取本地文件
    data = open("MontyPythonAlbums.csv", "r").read().decode('utf-8')
    print type(data)

    data_file = StringIO(data) # 字符串转为io对象
    csv_reader = csv.reader(data_file)
    for row in csv_reader:
        print(row)
        """
        ['Name', 'Year']
        ["Monty Python's Flying Circus", '1970']
        """

# readerCsv1()

# 按行读取列表
def readerCsv2():
    f = open("MontyPythonAlbums.csv", "r")
    csv_reader = csv.reader(f)
    for row in csv_reader:
        print(row)
        """
        ['Name', 'Year']
        ["Monty Python's Flying Circus", '1970']
        """
    f.close()

# readerCsv2()

# 按行读取字典,第一行为key
def readerCsv3():
    f = open("MontyPythonAlbums.csv", "r")
    csv_reader = csv.DictReader(f)
    for row in csv_reader:
        print(row)
        # {'Name': "Monty Python's Flying Circus", 'Year': '1970'}
    f.close()

# readerCsv3()