Python:如何 reshape 具有不同维度的多维数组？

Question

假设我有一个像这样的数组

[[1,2],[3,4,5]]

我想将其 reshape 为

[[[1],[2]],[[3],[4],[5]]]

Python 有没有一种简单的方法可以做到这一点？我知道如果整个数据的第二个维度都相同，这非常简单，但在我的例子中，第二个维度的长度分别是 2 和 3。

非常感谢。

Answer 1

如果您有这样的列表:

nested = [[1,2], [3,4,5]]

你可以像这样把它分开:

nested_split = [[[single_elt] for single_elt in inside_list] for inside_list in nested]

调用 print 时会得到以下输出:

[[[1], [2]], [[3], [4], [5]]]

内部或外部列表的维数不会以任何方式影响此解决方案，因为使用 for 循环和列表理解将动态适应任何大小的列表。