diff --git a/snippets/spread.md b/snippets/spread.md index 1f2957cf0..982f67470 100644 --- a/snippets/spread.md +++ b/snippets/spread.md @@ -11,13 +11,10 @@ Loop over elements, use `list.extend()` if the element is a list, `list.append() def spread(arg): ret = [] for i in arg: - if isinstance(i, list): - ret.extend(i) - else: - ret.append(i) + ret.extend(i) if isinstance(i, list) else ret.append(i) return ret ``` ```py -spread([1,2,3,[4,5,6],[7],8,9]) # [1,2,3,4,5,6,7,8,9] -``` \ No newline at end of file +spread([1, 2, 3, [4, 5, 6], [7], 8, 9]) # [1, 2, 3, 4, 5, 6, 7, 8, 9] +```