编写Python代码以使用条件语句从现有列创建新的CSV列。
import csv
# 读取原始CSV文件
with open('input.csv', 'r') as file:
reader = csv.reader(file)
rows = list(reader)
# 添加新的列名
rows[0].append('新列名')
# 根据条件创建新列的值
for row in rows[1:]:
# 根据现有列的值判断条件,并设置新列的值
if row[0] == '条件1':
row.append('条件1的结果')
elif row[0] == '条件2':
row.append('条件2的结果')
else:
row.append('其他条件的结果')
# 写入包含新列的CSV文件
with open('output.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(rows)
代码说明:
csv
模块打开原始CSV文件,并将其读取为一个列表形式的rows
变量。rows
的第一行(索引为0)的末尾添加了新的列名。rows
列表中的每一行(从第二行开始),根据现有列的值判断条件,并设置新列的值。csv
模块将包含新列的rows
列表写入到输出CSV文件。这段代码用于根据现有列的条件创建新列的值。在代码中,我们使用了条件语句(if-elif-else)来判断现有列的值,并根据不同的条件设置新列的值。根据实际需求,你可以根据不同的条件编写自己的条件语句。
请注意,代码中的文件名是示例文件名。你需要根据实际情况将input.csv
替换为你要读取的原始CSV文件名,将output.csv
替换为你要写入的输出CSV文件名。
关于腾讯云相关产品和产品介绍的链接地址,可以根据具体需要在腾讯云官方网站上查找相关信息。
领取专属 10元无门槛券
手把手带您无忧上云