Python:如何将 Numpy 数组项值类型转换为具有该值的数组？

Question

转换此类 ndarray 的最佳方法是什么:

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

至:

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

只需将每个值包装到数组

Answer 1

您可以使用 np.newaxis/None 引入新轴最后，就像这样 -

arr[...,None]

示例运行 -

In [6]: arr = np.array([[1,2,3], [4,5,6]])

In [7]: arr[...,None]
Out[7]: 
array([[[1],
        [2],
        [3]],

       [[4],
        [5],
        [6]]])

In [8]: arr[...,None].tolist() # To show it as a list for expected o/p format
Out[8]: [[[1], [2], [3]], [[4], [5], [6]]]