在 Matlab 中，如果您想在给定某些值条件的情况下获取向量中的值的子集，请执行以下操作:
negative_values = vec(vec<0)
positive_values = vec(vec>0)
我目前正在使用自制函数在 Python 中执行此操作，但这有点繁重。是否有更优雅的方式进行或我不知道的标准功能？我希望能够简洁地做类似的事情
negative_values = val.index(val<0)
positive_values = val.index(val>0)
但显然这不适用于 list.index()，因为它不应该获取表达式作为参数。
请您参考如下方法:
你可以像这样使用列表理解作为过滤器
numbers = [-10, -9, -8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
negatives = [number for number in numbers if number < 0]
print negatives
# [-10, -9, -8, -7, -6, -5, -4, -3, -2, -1]
positives = [number for number in numbers if number >= 0]
print positives
# [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
或者，您可以使用 filter函数，像这样
negatives = filter(lambda number: number < 0, numbers)
positives = filter(lambda number: number >= 0, numbers)
