爬虫(11)——csv的读入写入

最新推荐文章于 2024-06-20 21:17:30 发布

原创最新推荐文章于 2024-06-20 21:17:30 发布 · 854 阅读

0 ·

CC 4.0 BY-SA版权

爬虫同时被 2 个专栏收录

18 篇文章

订阅专栏

python

17 篇文章

订阅专栏

本文详细介绍了如何使用Python的csv模块进行CSV文件的读写操作，包括以元组和字典形式写入数据，以及如何从CSV文件中读取特定列的数据。

写入csv文件

#以元组的形式写入
def write_csv1():
    header=["username","age","height"]
    values=[("张三","18","180"),
            ("李四","19","190"),
            ("孙五","20","160")]
    with open("abc.csv",'w',encoding='utf-8',newline='') as fp:#newline=''是为了防止行与行之间有空格，默认为'\n '
        writer=csv.writer(fp)
        writer.writerow(header)#把表头读入
        writer.writerows(values)

#以字典的形式写入
def write_csv2():
    header=["username","age","height"]
    values=[{"name":"张三","Age":"18","height":"180"},
            {"name": "李四", "Age": "19", "height": "190"},
            {"name": "孙五", "Age": "20", "height": "160"}]
    with open("abc.csv",'w',encoding='utf-8',newline='') as fp:#newline=''是为了防止行与行之间有空格，默认为'\n '
        writer=csv.DictWriter(fp,header)
        writer.writeheader()#把表头读入
        writer.writerows(values)

爬虫的读出：

def read_csv1():
    with open("abc.csv", "r") as fp:
        reader = csv.reader(fp)  # reader是一个迭代器
        next(reader)  # 不想读取第一行
        for x in reader:
            name = x[1]  # 取第2列
            volumn = x[-1]  # 去最后一列
            print({"name:": name, "volumn": volumn})


def read_csv2():
    with open("文件的名字","r") as fp:
        #reader是一个迭代器，是以字典的形式读出来
        reader=csv.DictReader(fp)
        for x in reader:
            value={"name":x["name"],"volumn":x["volumn"]}
            print(value)