use set()
This commit is contained in:
@ -2,11 +2,12 @@
|
||||
|
||||
Returns the difference between two iterables.
|
||||
|
||||
Use list comprehension to only keep values not contained in `b`
|
||||
Use list comprehension to only keep values not contained in `b`. Use set to test `in` in `O(1)` time.
|
||||
|
||||
```python
|
||||
def difference(a, b):
|
||||
return [item for item in a if item not in b]
|
||||
_b = set(b)
|
||||
return [item for item in a if item not in _b]
|
||||
```
|
||||
``` python
|
||||
difference([1, 2, 3], [1, 2, 4]) # [3]
|
||||
|
||||
Reference in New Issue
Block a user