use set()
This commit is contained in:
@ -2,11 +2,12 @@
|
|||||||
|
|
||||||
Returns the difference between two iterables.
|
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
|
```python
|
||||||
def difference(a, b):
|
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
|
``` python
|
||||||
difference([1, 2, 3], [1, 2, 4]) # [3]
|
difference([1, 2, 3], [1, 2, 4]) # [3]
|
||||||
|
|||||||
Reference in New Issue
Block a user