要使用Python处理CSV格式的数据,可以使用内置的csv模块。以下是一个简单的示例,展示了如何读取、写入和操作CSV文件:
import csv
# 读取CSV文件
with open('data.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
# 对每一行进行操作
print(row)
# 写入CSV文件
data = [
['Name', 'Age', 'City'],
['John', '25', 'New York'],
['Alice', '30', 'London'],
['Bob', '35', 'Paris']
]
with open('output.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(data)
# 追加到CSV文件
new_data = ['Tom', '40', 'Berlin']
with open('output.csv', 'a', newline='') as file:
writer = csv.writer(file)
writer.writerow(new_data)
# 使用字典方式读取CSV文件
with open('data.csv', 'r') as file:
reader = csv.DictReader(file)
for row in reader:
# 对每一行进行操作
print(row['Name'], row['Age'], row['City'])
上述示例中,
需要注意的是,在读取或写入CSV文件时,我们使用了with open() as file语句,它能确保在完成操作后自动关闭文件。
另外,我们还展示了如何使用csv.DictReader()函数通过字典方式读取CSV文件。在这种模式下,CSV文件的首行被当作字典的键,每一行的数据作为对应键的值。
请根据需要进行相应的修改和扩展来处理你的CSV文件。
这些功能都已经在小蜜蜂AI上线,网址:https://zglg.work