Структурированный массив Numpy Mean
Предположим, что у меня есть структурированный массив студентов (строки) и результатов тестов (целых), где каждая запись - это результат, полученный конкретным студентом в ходе определенного теста. Естественно, у каждого ученика есть несколько записей в этом массиве.
Exampleimport numpy
grades = numpy.array([('Mary', 96), ('John', 94), ('Mary', 88), ('Edgar', 89), ('John', 84)],
dtype=[('student', 'a50'), ('score', 'i')])
print grades
#[('Mary', 96) ('John', 94) ('Mary', 88) ('Edgar', 89) ('John', 84)]
Как мне легко вычислить средний балл каждого ученика? Другими словами, как мне взять среднее значение массива в «баллах»? измерение? Я хотел бы сделать
grades.mean('score')
и вернуть Numpy
[('Mary', 92), ('John', 89), ('Edgar', 89)]
но Numpy жалуется
TypeError: an integer is required
Есть ли Numpy-esque способ сделать это легко? Я думаю, что это может включать просмотр структурированного массива с другим dtype. Любая помощь будет оценена. Благодарю.
Edit>>> grades = numpy.zeros(5, dtype=[('student', 'a50'), ('score', 'i'), ('testid', 'i'])
>>> grades[0] = ('Mary', 96, 1)
>>> grades[1] = ('John', 94, 1)
>>> grades[2] = ('Mary', 88, 2)
>>> grades[3] = ('Edgar', 89, 1)
>>> grades[4] = ('John', 84, 2)
>>> np.mean(grades, 'testid')
TypeError: an integer is required