列表中元素的总和

Question

我的 csv 数据看起来像这样：

ID, DATE, mm

251691,01/01/2016,16.6
251691,02/01/2016,4.4
251691,08/01/2016,3.7
120530,07/01/2019,55.5
120530,22/04/2019,1.8

我想把每年所有的“mm”和 ID 相加

我想要得到的结果是这样的：

('251691', '2016')  :  (sum_of_mm_for_each_station_in_that_year, number_of_total_rain_days)

那个：

('251691', '2016')  :  (24.7, 3)

这是我的代码：

answer = {}
with open(filename) as f:
        header = line_to_list(f.readline())
        if header[0] != 'ID':
            raise Exception("Bad")
        for line in f.readlines():
            row = line_to_list(line)
            date = row[1]
            mm = row[2]
            year = date [-4:]
            answer[row[0],year] = (mm, date)

函数 line_to_list：

def line_to_list(line):
    ''' converts a csv line (string) to a list of items '''
    line = line.rstrip('\n')
    return [s for s in line.split(',')]

我无法对元素求和。 我想每年按 ID 分组并对元素求和，但我不能使用 pandas 等任何模块。

有人可以帮我吗？ 谢谢！

Answer 1

我已对代码进行了一些更改以适应您的更改：-

answer = {}
def line_to_list(line):
    ''' converts a csv line (string) to a list of items '''
    line = line.rstrip('\n')
    return [s for s in line.split(',')]
with open("file.txt") as f:
    header = line_to_list(f.readline())
    if header[0] != 'ID':
        raise Exception("Bad")
    for line in f.readlines():
        row = line_to_list(line)
        date = row[1]
        mm = float(row[2])
        year = date [-4:]
        count = 1
        if (row[0], year) in answer:
            mm += answer[(row[0],year)][0]
            count += answer[(row[0],year)][1]
        answer[(row[0],year)] = (mm, count)
print(answer)

结果：-

{('251691', '2016'): (24.7, 3), ('120530', '2019'): (57.3, 2)}

解释：-

由于列表是可变的和可散列的，我们不能直接用它来引用它作为字典的键，因为键需要被散列。所以我明确地将它转换为元组，然后使用它。

代码if (row[0], year) in answer: 的部分将检查元组是否已经存在于字典中，如果存在，它将附加其值并使用mm += answer[(row[0],year)][0] 和count += answer[(row[0],year)][1] 行增加计数。我认为代码的另一部分很清楚。