csv模块简单使用

最新推荐文章于 2024-09-14 08:09:55 发布

Brave_liberty

最新推荐文章于 2024-09-14 08:09:55 发布

阅读量973

点赞数

分类专栏： python

本文链接：https://blog.youkuaiyun.com/qq_31396185/article/details/79632295

版权

python 专栏收录该内容

6 篇文章

订阅专栏

这篇博客介绍了如何使用Python的csv模块进行CSV文件的操作，包括读取CSV数据为列表和字典形式，特别适合处理网页表格数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

json是一种嵌套了列表与字典的格式，json包可以读取返回的json格式，json.load(html返回的对象)

csv模块，用来操作csv文件

import csv
#from os import open

csvFile = open("../files/test.csv", 'w+', newline='')
try:
    writer = csv.writer(csvFile)
    writer.writerow(('number', 'number plus 2', 'number times 2'))
    for i in range(10):
        writer.writerow( (i, i+2, i*2))
finally:
    csvFile.close()

爬取一个页面上的表格

import csv
from urllib.request import urlopen
from bs4 import BeautifulSoup

html = urlopen("http://en.wikipedia.org/wiki/Comparison_of_text_editors")
bsObj = BeautifulSoup(html, "html.parser")
#The main comparison table is currently the first table on the page
table = bsObj.findAll("table",{"class":"wikitable"})[0]
rows = table.findAll("tr")

csvFile = open("files/editors.csv", 'wt', newline='', encoding='utf-8')
writer = csv.writer(csvFile)
try:
    for row in rows:
        csvRow = []
        for cell in row.findAll(['td', 'th']):
            csvRow.append(cell.get_text())
        writer.writerow(csvRow)
finally:
    csvFile.close()

from urllib.request import urlopen
from io import StringIO
import csv

data = urlopen("http://pythonscraping.com/files/MontyPythonAlbums.csv").read().decode('ascii', 'ignore')
dataFile = StringIO(data)
csvReader = csv.reader(dataFile)

for row in csvReader:
    print("The album \""+row[0]+"\" was released in "+str(row[1]))

上边是读取结果为列表形式，下边用dictReader（）读取城字典形式

from urllib.request import urlopen
from io import StringIO
import csv

data = urlopen("http://pythonscraping.com/files/MontyPythonAlbums.csv").read().decode('ascii', 'ignore')
dataFile = StringIO(data)
dictReader = csv.DictReader(dataFile)

print(dictReader.fieldnames)

for row in dictReader:
    print(row)