python takes list and returns only if negative value also exists using set

Question

Basically I have a big list:

# where (n) is over a couple hundred thousand or is 1 million
def big_list(n):
    return [ randrange(-n//3,n//3) for i in range(n) ]

And using a set I must return a new list if and only if its negative value also exists.

Ex. if list = [-3,-2,-1,2,1,4] it should return new_list = [2,1]

I must do this using set, and I''m really lost.

Answer 1

Store all negative numbers in a set and then get the intersection with the initial list:

negatives = set(-x for x in data if x < 0)
numbers_with_negatives = negatives.intersection(data)

Demo:

>>> data
[-3, -2, -1, 2, 1, 4]
>>> negatives = set(-x for x in data if x < 0)
>>> negatives
set([1, 2, 3])
>>> negatives.intersection(data)
set([1, 2])