获取 CSV 列的平均值

在数据分析和处理过程中，获取 CSV 列的平均值是非常常见的需求。本文将介绍如何使用 Python 语言实现这个功能。

读取 CSV 文件

首先，需要读取 CSV 文件并将数据存储为列表。可以使用 Python 内置的 csv 模块来实现这个功能。以下是示例代码：

import csv

data = []
with open('filename.csv', newline='') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        data.append(row)

提取数据列

读取 CSV 文件后，需要从中提取指定的数据列。假设需要获取第二列数据，可以使用以下代码：

col = 1  # 第二列数据
column_data = [float(row[col]) for row in data[1:]]

以上代码使用列表解析语法从列表中提取第二列数据，使用 float() 函数将数据转换为浮点数并存储在另一个列表 column_data 中。

计算平均值

最后，可以使用 Python 内置的 statistics 模块计算列表中数值的平均值。以下是示例代码：

import statistics

mean = statistics.mean(column_data)
print(mean)

完整代码

将以上代码片段组合起来，可以得到完整的 Python 代码：

import csv
import statistics

data = []
with open('filename.csv', newline='') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        data.append(row)

col = 1  # 第二列数据
column_data = [float(row[col]) for row in data[1:]]

mean = statistics.mean(column_data)
print(mean)

总结

本文介绍了如何使用 Python 语言获取 CSV 列的平均值。通过使用 csv 模块读取 CSV 文件，使用列表解析提取数据列，使用 statistics 模块计算平均值，可以快速、准确地完成数据分析任务。