python - 在 python 中对数组使用多个索引

Question

我有一个关于如何为数组或 rec.array 使用多个索引的简单问题。更具体地说，我想隔离满足多个规定的数组中的单元格。例如:

import numpy as np
test = np.ones(5)
test_rec = test.view(recarray)
test_rec.age = np.array([0,1,2,1,4])
test_rec.sex = np.array([0,1,1,0,0])

我想隔离 test_rec，其中 test_rec 年龄为 1 且 test_rec.sex 为 1，即:

test_rec[test_rec.age==1 and test_rec.sex==1]

不幸的是，这不起作用。

Answer 1

使用 logical_and() 或 bitwise_and()，您可以使用 & 运算符来执行 bitwise_and():

test_rec[(test_rec.age==1) & (test_rec.sex==1)]

括号很重要，因为 & 的优先级低于 ==。