小编End*_*rju的帖子

如何提取numpy结构化数组中满足某些条件的行?

假设我在 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)

python numpy

1
推荐指数
1
解决办法
590
查看次数

标签 统计

numpy ×1

python ×1