Question

我有一个包含100列的csv文件。我想计算第4列到第n列的总和。我可以为单个列生成总和，但是当我为所有列尝试它时，我失败了。这是我到目前为止所拥有的

import decimal
import numpy as np
import os as os
import csv as csv
import re as re
import sys

col=10
values=[]
with open('test.csv', 'r') as f:
    reader = csv.reader(f)
    headers = reader.next()
    for line in reader:
    #print line
        line = [int(i) for i in line]
    col_totals = [sum(result) for result in zip(*line)]
    print col_totals
       #values.append(int(line[col]))
       #csum=sum(values)
    #print csum

谢谢，

Answer 1

这在熊猫中非常非常容易：

import pandas as pd
df = pd.read_csv(filename)
df[df.columns[4:]].sum()

如果你想要列的每行总和，那就是：

df[df.columns[4:]].sum(1)

Answer 2

如果你想在连续的行之间求和，那就可以了

i, j = 3, 5

with open('test.csv', 'r') as f:
    reader = csv.reader(f)
    headers = reader.next()
    table = list(reader)
    sums = [sum(float(elt) for elt in col) for col in zip(*table)[i:j]]

还尝试以下

requested = [4, 7, 12, 13, 21, 81]

with open('test.csv', 'r') as f:
    reader = csv.reader(f)
    headers = reader.next()
    table = list(reader)
    sums = [sum(float(elt) for elt in col) for i, col in enumerate(zip(*table)) if i in requested]

汇总CSV文件中的特定列

2 个答案: