Python入门：如何高效计算列表列总和及获取列长度？

高效计算平均值的方法 #

你当前逐个索引相加的方式只适合小数据集，针对百万级数据，推荐两种更高效、通用的实现方式：

1. 普通for循环（直观易理解） #

total_salary = 0
valid_rows = 0

for row in hr_data:
    # 先判断该行是否包含目标列（索引3），避免索引越界报错
    if len(row) >= 4:
        total_salary += row[3]
        valid_rows += 1

if valid_rows > 0:
    average_rate = total_salary / valid_rows
    print(f"The average salary of the employees is : {average_rate}")
else:
    print("没有有效的薪资数据")

2. 内置函数结合生成器（效率更高） #

Python内置的sum()底层基于C实现，处理大数据的速度比纯Python循环更快；搭配生成器表达式还能节省内存，无需一次性生成所有薪资的列表：

# 生成器表达式过滤出有效的薪资数据
valid_salaries = (row[3] for row in hr_data if len(row) >= 4)
total = sum(valid_salaries)
# 统计有效行数
valid_count = sum(1 for row in hr_data if len(row) >= 4)

if valid_count > 0:
    average_rate = total / valid_count
    print(f"The average salary of the employees is : {average_rate}")

获取列长度的方法 #

Python没有专门的"列长度"函数，但可以根据数据集的实际情况灵活处理：

• 如果所有行的列数一致：直接取任意一行的长度即可，比如column_num = len(hr_data[0])（注意要确保hr_data不为空）
• 如果行的列数不一致：
  • 获取最大列数：max_col = max(len(row) for row in hr_data)
  • 获取最小列数：min_col = min(len(row) for row in hr_data)
• 若要统计某一列的有效元素数量（比如你要计算的薪资列），就是上面代码里的valid_count变量。

内容的提问来源于stack exchange，提问作者Joshen Zu Xian