numpy利用布尔掩码做复杂索引

data=[(‘Alice’, 25, 55.0) (‘Bob’, 45, 85.5) (‘Cathy’, 37, 68.0)(‘Doug’, 19, 61.5)]
data[‘age’] < 30 #此布尔操作返回bool数组。#Output: array([ True, False, False, True])
data[data[‘age’]<30]#以布尔数组中真值索引标签。 #Output:array([(‘Alice’, 25, 55. ), (‘Doug’, 19, 61.5)])
data[data[‘age’] < 30][‘name’]. #Output:array([‘Alice’, ‘Doug’].第二层索引。
更复杂的索引功能要用pandas的DataFrame对象

猜你喜欢

转载自blog.csdn.net/qq_42422981/article/details/83119234