python - 在 Python 中是否有一个基于值条件构建子列表的简单功能？

Question

在 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()，因为它不应该获取表达式作为参数。

Answer 1

你可以像这样使用列表理解作为过滤器

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)