用小蜜蜂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