python怎么利用代码实现,要csv文件
把特定几列的数据加在一起并把总和放在新的一个列里面
看你csv具体内容,你调一下。没header就改成index。
import csv
target_columns = ["col1", "col3"]
with open("./has_header.csv") as rf, open("./has_header_res.csv", "w", newline='') as wf:
lines = rf.readlines()
rows = []
for row in csv.DictReader(lines):
row_dict = dict(row)
total = sum([int(row_dict[k]) for k in target_columns])
row_dict["total"] = total
rows.append(row_dict)
writer = csv.DictWriter(wf, fieldnames=rows[0].keys())
writer.writeheader()
for row in rows:
writer.writerow(row)
【以下回答由 GPT 生成】
可以使用pandas库来处理CSV文件,并使用Python编写脚本来实现该功能。以下是一个示例代码:
pip install pandas
read_csv()
函数读取CSV文件。import pandas as pd
# 读取CSV文件
df = pd.read_csv('your_file.csv')
sum()
函数计算需要相加的特定列的总和,并将结果存放在一个新的列中。# 选择需要相加的特定列
col1 = df['column1']
col2 = df['column2']
# 计算总和并将结果存放在新的列中
sum_col = col1 + col2
df['sum_column'] = sum_col
to_csv()
函数将修改后的数据保存到一个新的CSV文件中。# 保存修改后的数据到新的CSV文件
df.to_csv('your_new_file.csv', index=False)
完整的代码如下所示:
import pandas as pd
# 读取CSV文件
df = pd.read_csv('your_file.csv')
# 选择需要相加的特定列
col1 = df['column1']
col2 = df['column2']
# 计算总和并将结果存放在新的列中
sum_col = col1 + col2
df['sum_column'] = sum_col
# 保存修改后的数据到新的CSV文件
df.to_csv('your_new_file.csv', index=False)
请将代码中的"your_file.csv"替换为你实际的CSV文件名,并根据需要修改列名和文件名。
【相关推荐】