对csv文件的某一列进行分两组并求均值
时间: 2024-01-19 15:04:45 浏览: 62
假设你有一个名为`data.csv`的CSV文件,其中包含多列数据,你需要对其中一列进行分组并计算平均值。以下是一个示例代码:
```python
import csv
filename = "data.csv"
group_column = "group_column_name" # 需要分组的列的名称
value_column = "value_column_name" # 需要计算平均值的列的名称
group1_values = []
group2_values = []
with open(filename, "r") as file:
reader = csv.DictReader(file)
for row in reader:
if row[group_column] == "group1":
group1_values.append(float(row[value_column]))
elif row[group_column] == "group2":
group2_values.append(float(row[value_column]))
group1_mean = sum(group1_values) / len(group1_values)
group2_mean = sum(group2_values) / len(group2_values)
print("Group 1 mean: ", group1_mean)
print("Group 2 mean: ", group2_mean)
```
该代码将打开名为`data.csv`的文件并读取其中的数据。然后,它将根据`group_column_name`列中的值将数据分为两组。在本例中,我们将分为`group1`和`group2`。对于每个组,代码将从`value_column_name`列中获取所有值,并计算它们的平均值。最后,代码将打印出每个组的平均值。
阅读全文