在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()是行不通的，因为它不支持接收表达式作为参数。

相关问题：

• NumPy中最优雅的MATLAB "vec" 函数实现
• 在Python/Numpy中选择满足条件的特定行，且只选择某些列
• 获取二维数组中特定值的位置
• Python中类似MATLAB的find()函数
• 在Python中类似MATLAB的切片
• 高效查找Python关联列表中的元素
• 如何根据另一个DataFrame中的值在pandas DataFrame中创建新列
• 如何在Python中根据对应列的值排除数组中的值？

Answer 1

你需要使用 numpy，它是一个可以替代 matlab 的工具：

In [1]: import numpy as np

In [2]: a = np.arange(-5, 5)

In [3]: a
Out[3]: array([-5, -4, -3, -2, -1,  0,  1,  2,  3,  4])

In [4]: a[a>0]
Out[4]: array([1, 2, 3, 4])

In [5]: np.where(a>0)  #used to find the indices where the condition matches
Out[5]: (array([6, 7, 8, 9]),)

In [6]: np.where(a%2==0)
Out[6]: (array([1, 3, 5, 7, 9]),)

Answer 2

你可以用这种语法来搭配 numpy 使用：

import numpy

a = numpy.arange(10)
--> a = array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

a[a > 5]
--> array([6, 7, 8, 9])

Answer 3

你可以用列表推导式来做过滤，像这样

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)