📜  Python|元组列表第K列的求和(1)

📅  最后修改于: 2023-12-03 15:04:24.333000             🧑  作者: Mango

Python | 元组列表第K列的求和

在Python中,可以使用元组列表(也称为嵌套列表)来组织和存储数据。如果需要对这些数据进行汇总或分析,一种常见的需求是对其中的某一列进行求和。本文将介绍如何使用Python实现这一功能。

示例数据

我们先来定义一个示例数据,它是一个包含多个元组的列表。每个元组包含三个数字,分别表示学生的学号、年龄和成绩:

data = [
    (1, 18, 90),
    (2, 19, 89),
    (3, 18, 87),
    (4, 19, 92),
    (5, 20, 95)
]
求和函数

接下来,我们定义一个函数,它可以对元组列表的第K列求和。具体实现如下:

def sum_col(lst, k):
    return sum(row[k] for row in lst)

其中,lst表示待汇总的元组列表,k表示要对哪一列进行求和。函数的实现非常简单,它使用了列表推导式来遍历所有元组,然后对它们的第K列求和。

测试

我们可以对上述示例数据进行测试,验证函数的正确性。例如,下面的代码会输出成绩的总和:

total_score = sum_col(data, 2)
print(total_score)  # 输出:453

同理,如果需要求年龄或学号的总和,只需要修改函数调用中的第二个参数即可。例如,下面的代码会输出年龄的总和:

total_age = sum_col(data, 1)
print(total_age)  # 输出:94
总结

本文介绍了如何使用Python对元组列表的某一列进行求和。这个过程可以使用列表推导式来实现,非常简单易懂。如果读者还有其他关于Python编程的问题或需求,可以在下方评论区留言,我们将尽快为您解答。