python - 如何在 python 中访问 numpy 结构化数组中的多个列值？

Question

我是 numpy 的新手，仍在学习它。我创建了一个结构化数组，如下所示：

name = ['Alice', 'Beth', 'Cathy', 'Dorothy']
studentid = [1,2,3,4]
score = [85.4, 90.4, 87.66, 78.9]
student_data = np.zeros(4, dtype={'names':('name', 'studentId', 'score'), 
                         'formats':('U10', 'i4', 'f8')})
student_data['name'] = name
student_data['studentId'] = studentid
student_data['score']=score

为了得到分数大于 85 的人的名字，我写了这个：

student_data[student_data['score'] > 85]['name']

但是，如果我尝试检索另一列以及“名称”，则会收到错误消息：

IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices

例如，我尝试了以下方法：

student_data[student_data['score'] < 90]['name','studentid']
student_data[student_data['score'] < 90]['name']['studentid']

它们都导致错误：

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-189-7ad7c151f0e3> in <module>
----> 1 student_data[student_data['score'] < 90]['name']['studentid']

IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices

谁能让我知道我在这里犯了什么错误以及如何在有条件的基础上检索多个列？

标签： pythonnumpy

Answer 1

您可以使用列数组访问：

student_data[student_data['score'] < 90][['name','studentid']]

输出：

array([('Alice', 1), ('Cathy', 3), ('Dorothy', 4)],
      dtype={'names':['name','studentId'], 'formats':['<U10','<i4'], 'offsets':[0,40], 'itemsize':52})