numpy利用布尔掩码做复杂索引
1 data=[(‘Alice’, 25, 55.0) (‘Bob’, 45, 85.5) (‘Cathy’, 37, 68.0)(‘Doug’, 19, 61.5)] 2 3 data[‘age’] < 30 #此布尔操作返回bool数组。 4 #Output: 5 array([ True, False, False, True]) 6 7 data[data[‘age’]<30]#以布尔数组中真值为索引标签。 #Output: 8 array([(‘Alice’, 25, 55. ), (‘Doug’, 19, 61.5)]) 9 10 data[data[‘age’] < 30][‘name’]. 11 #Output: 12 array([‘Alice’, ‘Doug’].第二层索引。
更复杂的索引功能要用pandas的DataFrame对象。