flatten

jackzhenguo · jackzhenguo · commit b82ebd811ae3 · 2019-11-19T08:12:17.000+08:00
diff --git a/flatten.md b/flatten.md
@@ -0,0 +1,48 @@
+例子卡片7：嵌套数组完全展开
+
+对于如下数组：
+
+```
+[[[1,2,3],[4,5]]]
+```
+
+如何完全展开成一维的。这个小例子实现的`flatten`是递归版，两个参数分别表示带展开的数组，输出数组。
+
+```python
+from collections.abc import *
+
+# 返回list
+def flatten(input_arr, output_arr=None):
+    if output_arr is None:
+        output_arr = []
+    for ele in input_arr:
+        if isinstance(ele, Iterable): # 判断ele是否可迭代
+            flatten(ele, output_arr)  # 尾数递归
+        else:
+            output_arr.append(ele)    # 产生结果
+    return output_arr
+```
+
+调用`flatten`:
+
+```
+print(flatten([[1,2,3],[4,5]]))
+print(flatten([[1,2,3],[4,5]], [6,7]))
+print(flatten([[[1,2,3],[4,5,6]]]))
+# 结果：
+[1, 2, 3, 4, 5]
+[6, 7, 1, 2, 3, 4, 5]
+[1, 2, 3, 4, 5, 6]
+```
+
+numpy里的`flatten`与上面的函数实现有些微妙的不同：
+
+```
+import numpy
+b = numpy.array([[1,2,3],[4,5]])
+b.flatten()
+array([list([1, 2, 3]), list([4, 5])], dtype=object)
+```
+
+你看还是与我们小例子中的不太一样吧。
+
