以下是一个使用条件语句从现有列创建一个新的CSV列的Python代码示例:
import csv
def create_new_column(input_file, output_file):
with open(input_file, 'r') as file:
reader = csv.reader(file)
header = next(reader) # 获取CSV文件的头部
# 添加新的列名到头部
header.append('New Column')
with open(output_file, 'w', newline='') as output:
writer = csv.writer(output)
writer.writerow(header) # 写入头部到输出文件
for row in reader:
value = row[0] # 假设需要根据第一列的值创建新的列
# 根据条件创建新的列值
if value == 'A':
new_column_value = 'Value A'
elif value == 'B':
new_column_value = 'Value B'
else:
new_column_value = 'Other Value'
row.append(new_column_value) # 将新的列值添加到行
writer.writerow(row) # 写入行到输出文件
# 示例使用
input_file = 'input.csv'
output_file = 'output.csv'
create_new_column(input_file, output_file)
上述代码中,我们首先打开输入CSV文件并读取其内容。接着,在头部添加一个新的列名,并将其写入输出文件。然后,我们逐行读取输入文件的内容,并根据特定条件创建新的列值。最后,将新的列值添加到行中,并将整行写入输出文件。
请确保将input.csv
替换为你的输入文件的路径,并将output.csv
替换为你想要生成的输出文件的路径。