python - N维数组变成单个数组？

Question

我有一个列表数组，我想将其转换为一维数组。我可以用 2 个“for”循环来做到这一点，但一个禁令是我必须这样做，因为它是 N 维数组。

original_list = [[2,4,3,[1,2]],[1,5,6], [9], [7,9,0]]

Answer 1

只需chain使用itertools：

代码：

import itertools as it

original_list = [[2,4,3,[1,2]],[1,5,6], [9], [7,9,0]]

print(original_list)

new_list = list(it.chain.from_iterable(original_list))

print(new_list)

输出：

[[2, 4, 3, [1, 2]], [1, 5, 6], [9], [7, 9, 0]]
[2, 4, 3, [1, 2], 1, 5, 6, 9, 7, 9, 0]