From 3217e7a085880221cea31ea4405c00f0013fdbdd Mon Sep 17 00:00:00 2001 From: Angelos Chalaris Date: Fri, 3 Jan 2020 12:58:28 +0200 Subject: [PATCH] Update code in spread --- snippets/spread.md | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) 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] +```