用小蜜蜂AI实现Python处理csv格式数据

要使用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.reader()函数用于读取CSV文件，并返回迭代器对象，可用于遍历所有行。csv.writer()函数用于创建一个写入器对象，通过调用其writerow()或writerows()方法来写入数据。

需要注意的是，在读取或写入CSV文件时，我们使用了with open() as file语句，它能确保在完成操作后自动关闭文件。

另外，我们还展示了如何使用csv.DictReader()函数通过字典方式读取CSV文件。在这种模式下，CSV文件的首行被当作字典的键，每一行的数据作为对应键的值。

请根据需要进行相应的修改和扩展来处理你的CSV文件。

这些功能都已经在小蜜蜂AI上线，网址：https://zglg.work