从包含特定字符的 CSV 列中查找平均值答案

【问题标题】：find average value from CSV columns that contain a specific character从包含特定字符的 CSV 列中查找平均值
【发布时间】：2015-09-21 17:11:16
【问题描述】：

我正在尝试获取一个简单的 python 函数，该函数将读取 CSV 文件并找到列和行的平均值。该函数将检查第一行和每一列的标题以字母“Q”开头，它将计算值的平均值该列，然后将其打印到屏幕上。然后对于每一行数据它将计算列中所有项目的学生平均值以“Q”开头。它会正常计算这个平均值最低的测验被放弃了。它将为每个学生打印出两个值。

CSV 文件包含学生的成绩，如下所示：

       hw1   hw2    Quiz3 hw4   Quiz2   Quiz1
john    87    98    76    67    90      56
marie   45    67    65    98    78      67
paul    54    64    93    28    83      98
fred    67    87    45    98    56      87

到目前为止我的代码是这样的，但我不知道如何继续：

import csv


def practice():
newlist=[]
afile= input('enter file name')
a = open(afile, 'r')
reader = csv.reader(a, delimiter = ",")


for each in reader:
    newlist.append(each)
y=sum(int(x[2] for x in reader))
print (y)

filtered = []
total = 0

for i in range (0,len(newlist)):
    if 'Q' in [i][1]:
        filtered.append(newlist[i])
return filtered

【问题讨论】：

您说您有一个 csv 文件，并且您的代码将分隔符指定为 ","，但您的示例文本似乎没有任何逗号。您是否出于某种原因删除了它们？

标签： python csv sum average

【解决方案1】：

我可以建议使用 Pandas：

>>> import pandas as pd
>>> data = pd.read_csv('file.csv', sep=' *')
>>> q_columns = [name for name in data.columns if name.startswith('Q')]

>>> reduced_data = data[q_columns].copy()
>>> reduced_data.mean()
Quiz3    69.75
Quiz2    76.75
Quiz1    77.00
dtype: float64

>>> reduced_data.mean(axis=1)
john     74.000000
marie    70.000000
paul     91.333333
fred     62.666667
dtype: float64

>>> import numpy as np
>>> for index, column in reduced_data.idxmin(axis=1).iteritems():
...     reduced_data.ix[index, column] = np.nan
>>> reduced_data.mean(axis=1)
john     83.0
marie    72.5
paul     95.5
fred     71.5
dtype: float64

【讨论】：

【解决方案2】：

如果您更改.csv 格式，您将获得更好的代码。这样我们就可以轻松使用DictReader了。

grades.csv：

name,hw1,hw2,Quiz3,hw4,Quiz2,Quiz1
john,87,98,76,67,90,56
marie,45,67,65,98,78,67
paul,54,64,93,28,83,98
fred,67,87,45,98,56,87

代码：

import numpy as np
from collections import defaultdict
import csv

result = defaultdict( list )
with open('grades.csv', 'r') as csvfile:
    reader = csv.DictReader(csvfile) 
    for row in reader:
        for k in row:
            if k.startswith('Q'):
                result[ row['name'] ].append( int(row[k]) )
for name, lst in result.items():
    print name, np.mean( sorted(lst)[1:] )

输出：

paul 95.5
john 83.0
marie 72.5
fred 71.5

【讨论】：