我是一名计算机科学老师,正在尝试使用NumPy为自己创建一本小写本.但我认为如果我可以创建一个使用行和列的字段名称的ndarray,它将使我的代码更容易编写.这是我到目前为止所得到的:
import numpy as np
num_stud = 23
num_assign = 2
grades = np.zeros(num_stud, dtype=[('assign 1','i2'), ('assign 2','i2')]) #etc
gv = grades.view(dtype='i2').reshape(num_stud,num_assign)
Run Code Online (Sandbox Code Playgroud)
所以,如果我的第一个学生在'assign 1'上得到97,我可以写下:
grades[0]['assign 1'] = 97
gv[0][0] = 97
Run Code Online (Sandbox Code Playgroud)
另外,我可以做以下事情:
np.mean( grades['assign 1'] ) # class average for assignment 1
np.sum( gv[0] ) # total points for student 1
Run Code Online (Sandbox Code Playgroud)
这一切都有效.但我无法弄清楚该怎么做是使用学生ID号来指代某个学生(假设我的两个学生有学生ID,如图所示):
grades['123456']['assign 2'] = 95
grades['314159']['assign 2'] = 83
Run Code Online (Sandbox Code Playgroud)
...或者可能使用不同的字段名称创建第二个视图?
np.sum( gview2['314159'] ) # total points for the student with the given id
Run Code Online (Sandbox Code Playgroud)
我知道我可以创建一个dict将学生ID映射到索引,但这似乎是脆弱和苛刻的,我希望有一个更好的方法:
id2i = { …Run Code Online (Sandbox Code Playgroud)