当前位置:首页python > 正文

python中给csv替换数据

作者:野牛程序员:2023-08-10 17:48:00python阅读 2933

在Python中,你可以使用csv模块或其他字符串处理方法来替换CSV文件中的数据。以下是使用csv模块进行替换数据的示例:

假设我们有一个名为 "data.csv" 的CSV文件,内容如下:

Name, Age, Country
Alice, 25, USA
Bob, 30, Canada
Charlie, 28, UK

现在我们将对年龄进行替换,将年龄小于30岁的人的年龄替换为"young",年龄大于等于30岁的人的年龄替换为"adult"。

以下是使用csv模块进行数据替换的示例:

import csv

def replace_age(age):
    age = int(age)
    if age < 30:
        return "young"
    else:
        return "adult"

def replace_csv_data(input_file, output_file):
    with open(input_file, 'r') as input_csv, open(output_file, 'w', newline='') as output_csv:
        csv_reader = csv.reader(input_csv)
        csv_writer = csv.writer(output_csv)

        csv_writer.writerow(next(csv_reader))  # 写入标题行

        for row in csv_reader:
            name, age, country = row
            new_age = replace_age(age)
            csv_writer.writerow([name, new_age, country])

input_csv_file = 'data.csv'
output_csv_file = 'data_modified.csv'
replace_csv_data(input_csv_file, output_csv_file)

在这个示例中,replace_age函数根据年龄的值返回"young"或"adult"。replace_csv_data函数读取原始CSV文件,替换年龄列的数据,并将替换后的数据写入新的CSV文件中。

运行这个代码后,新的CSV文件 "data_modified.csv" 将包含替换后的数据。

请注意,这只是一个简单的示例,实际中你可能需要根据具体情况进行更复杂的数据替换操作。


野牛程序员教少儿编程与信息学奥赛-微信|电话:15892516892
野牛程序员教少儿编程与信息学竞赛-微信|电话:15892516892
相关推荐

最新推荐

热门点击