假设我在 numpy 中创建结构化数组:
name = ['Tom' , 'Jim', 'Alice', 'Alice', 'Greg']
height = [188, 160, 160, 157, 180]
pet = ['dog', 'cat', 'fish', 'dog', 'cat']
a = np.zeros(len(name), dtype=[('name', 'U30'), ('height', 'i'), ('pet', 'U30')])
a['name'] = name
a['height'] = height
a['pet'] = pet
Run Code Online (Sandbox Code Playgroud)
numpy 有没有办法提取满足某些条件的行。例如:
'height' == 160 and 'pet' == 'cat'
Run Code Online (Sandbox Code Playgroud)