python - 返回列表中正数或负数元素的函数

Question

我正在尝试创建一个 splitList 函数，该函数将选项作为输入并返回列表的正元素或负元素。

到目前为止，这就是我所拥有的:

def splitList([1,-3,5,7,-9,-11,10,2,-4], option)

• 如果“option”= 0，我想获得所有积极的元素。
• 如果“option”= 1，我想获取所有否定元素。

Answer 1

这是一个稍微干燥的替代方案:

import operator

def split_list(iterator, positive=True):
    sign_test = operator.gt if positive else operator.lt
    return [i for i in iterator if sign_test(i, 0)]

如果您希望它返回非正/非负，您可以将 gt 和 lt 替换为 ge 和 le.