把特定几列的数据加在一起并把总和放在新的一个列里面

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)

img

【以下回答由 GPT 生成】

可以使用pandas库来处理CSV文件,并使用Python编写脚本来实现该功能。以下是一个示例代码:

  1. 首先,安装pandas库。在PyCharm的终端中输入以下命令:
pip install pandas
  1. 导入pandas库和读取CSV文件。在Python脚本中,导入pandas库并使用read_csv()函数读取CSV文件。
import pandas as pd

# 读取CSV文件
df = pd.read_csv('your_file.csv')
  1. 创建一个新列来存放总和。使用sum()函数计算需要相加的特定列的总和,并将结果存放在一个新的列中。
# 选择需要相加的特定列
col1 = df['column1']
col2 = df['column2']

# 计算总和并将结果存放在新的列中
sum_col = col1 + col2
df['sum_column'] = sum_col
  1. 保存修改后的数据到新的CSV文件。使用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文件名,并根据需要修改列名和文件名。



【相关推荐】



如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^