Python - 将整数列表拆分为正数和负数

时间:2015-03-15 07:06:53

标签: python python-3.x

我正在学习python,我想知道如何拆分列表如下:

A = [1, -3, -2, 8, 4, -5, 6, -7]

分成两个列表,一个包含正数,另一个包含负整数:

B = [1, 8, 4, 6]
C = [-3, -2, -5, -7]

6 个答案:

答案 0 :(得分:2)

您可以使用defaultdict()

在O(n)中执行此操作
In [3]: from collections import defaultdict

In [4]: d = defaultdict(list)

In [5]: for num in A:
   ...:     if num < 0:
   ...:         d['neg'].append(num)
   ...:     else: # This will also append zero to the positive list, you can change the behavior by modifying the conditions 
   ...:         d['pos'].append(num)
   ...:         

In [6]: d
Out[6]: defaultdict(<class 'list'>, {'neg': [-3, -2, -5, -7], 'pos': [1, 8, 4, 6]})

另一种方法是使用两个单独的列表推导(不推荐用于长列表):

>>> B,C=[i for i in A if i<0 ],[j for j in A if j>0]
>>> B
[-3, -2, -5, -7]
>>> C
[1, 8, 4, 6]

或使用filter功能:

In [19]: list(filter((0).__lt__,A))
Out[19]: [1, 8, 4, 6]

In [20]: list(filter((0).__gt__,A))
Out[20]: [-3, -2, -5, -7]

答案 1 :(得分:0)

如果你能负担两次迭代A,列表推导是最好的IMO:

B = [x for x in A if x >= 0]
C = [x for x in A if x < 0]

当然总是采用“手动”的方式:

A = [1, -3, -2, 8, 4, -5, 6, -7]
B = []
C = []
for x in A:
    if (x >= 0):
        B.append(x)
    else:
        C.append(x)

答案 2 :(得分:0)

我们可以使用一个函数将+ ve数和-ve数与一个for循环

分开
def separate_plus_minus(numbers_list):
    positive = []
    negative = []
    for number in numbers_list:
        if number >= 0:
             positive.append(number)
        else:
             negative.append(number)
    return positive, negative

usuage:

numbers_list = [-1,5,6,-23,55,0,25,-10]
positive, negative = separate_plus_minus(numbers_list)

输出:

positve = [5, 6, 55, 0, 25]
negative = [-1, -23, -10]

答案 3 :(得分:0)

def manipulate_data(alist):
fo = []
go = []
for i in alist:
    if i < 0:
        fo.append(i)
    #elif(i > 0):
    #   go.append(i)
print fo
for i in alist:
    if i > 0:
        go.append(i)
print go

def main():     alist = [54,26,-93,-17,-77,31,44,55,20]     manipulate_data(ALIST)

如果名称 =='主要':     main()的

这是另一种方法

答案 4 :(得分:0)

你也可以像这样在列表理解中使用条件来做到这一点:

k = [1, -3, 5, 9, -1, 3, -3, -2]
positive = list()
negative = [i for i in k if i < 0 or (i > 0 and positive.append(i))]

我很确定这是 O(n) 并且只有两行。

答案 5 :(得分:-1)

我们可以使用 lambda 如下:

list(filter(lambda x: 0 > x, A))
Output: [-3, -2, -5, -7]

list(filter(lambda x: 0 <= x, A))
Output: [1, 8, 4, 6]

我们必须将方程(0 <= x) 改为(0 < x) 以避免正数列表中的0。