python - Python 选择向量负分量的综合方法

Question

例如，如果我有一个像

import numpy as np
a=np.array([1,2,-3,4,-5])

如果我想创建一个b带有负值的向量a我当然可以做一个for循环

list=[]
for i in a:
    if i<0:
        list.append(i)
b=np.array(list)

但我确信有更好的方法。什么是更合成的、类似 Python 的方式来做同样的事情？

标签： python

Answer 1

数组本质上可以由布尔数组选择，因此：

a=np.array([1,2,-3,4,-5])
b=a[a<0]

给

>>> b
array([-3, -5])