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

- 上一篇:Python将csv字符串转换为数组
- 下一篇:python按列写入csv