如何在 python 中找到 csv 文件的列的平均值？

Question

我正在尝试从 csv 文件中不同时间的 APM 列表中查找平均 APM（每分钟操作数）。当我尝试使用此代码进行操作时：

import csv

with open('test_game.csv') as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
    x = (row['Total APM'])
    x_sum = sum(x)
    x_length = len(x)
    x_average = x_sum / x_length
    print(x_average)

我收到此错误：

Traceback (most recent call last):
File "C:/Users/Luke's Laptop/Desktop/magicka_practice.py", line 7, in <module>
x_sum = sum(x)
TypeError: unsupported operand type(s) for +: 'int' and 'str'

这是否意味着我必须将 csv 行的输出更改为整数列表（如果是这样怎么办？）还是我的代码存在明显错误。我对此很陌生，所以这可能是一个愚蠢的问题，或者我可能正在做一些非常明显的事情。我很感激能得到任何帮助。

Answer 1

改变这一行：

x = (row['Total APM'])

到：

x = int(row['Total APM'])

这会将您的字符串/数字转换为可以进行数值运算的实际整数。

以下是您的代码中可能发生的情况：

>>> x = "1"
>>> sum(x)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: unsupported operand type(s) for +: 'int' and 'str'
>>> y = int(x)
>>> sum(y)
1

Python 是strongly typed，因此通常您不能对不兼容的不同类型执行任意操作。例如：int+str 将导致 TypeError。

你可能想要的是这个：

import csv

with open('test_game.csv') as csvfile:
reader = csv.DictReader(csvfile)
xs = []
for row in reader:
    try:
        x = int(row['Total APM'])
        xs.append(x)
    except ValueError:
        print("Error converting: {0:s}".format(row['Total APM']))
x_average = sum(xs) / len(xs)
print(x_average)

注意：您希望在遍历所有行后计算整列的“平均值”。所以你想保持一个运行总数，然后在循环之外计算平均值。

更新：或者（更符合您的原始算法）您可以这样做：

导入 csv

with open('test_game.csv') as csvfile:
reader = csv.DictReader(csvfile)
x_sum = x_length = 0
for row in reader:
    try:
        x = row['Total APM']
        x_sum = += int(x)
        x_length += 1
    except ValueError:
        print("Error converting: {0:s}".format(x))
x_average = x_sum / x_length
print(x_average)

这会保持一个运行总和和总长度，但 *NB** 除非你想计算一个运行平均值，否则你仍然必须在循环之外计算平均值 :)

更新 #2： 正如@Karl 所建议的那样——捕捉任何错误并适当地处理它们是一个好的想法。 “适当”取决于您，取决于用例:)