存储数据
从之前代码的运行结果可以看到,我们是将爬取到的数据都储存到了新建的 csv 文件中。
所以我们可以使用 csv 模块的 DictWriter() 类,接下来我们简单回顾一下这个知识点。
1)调用 csv 模块中类 DictWriter 的语法为:csv.DictWriter(f, fieldnames)。语法中的参数 f 是 open() 函数打开的文件对象;参数 fieldnames 用来设置文件的表头;
2)执行csv.DictWriter(f, fieldnames)后会得到一个 DictWriter 对象;
3)得到的 DictWriter 对象可以调用 writeheader() 方法,将 fieldnames 写入 csv 的第一行;
4)最后,调用 writerows() 方法将多个字典写进 csv 文件中。
# 新建 csv 文件存储书籍信息
with open('books.csv', 'w') as f:
# 将文件对象转换成 DictWriter 对象
writer = csv.DictWriter(f, fieldnames=['书名', '作者', '出版社', 'ISBN', '页数', '出版年', '定价'])
# 写入表头与数据