将Numpy数组的元素转换为列表

Question

我有一个numpy数组，看起来可能像这样

[[1 2]
 [3 4]]

我想将其转换为一个额外的维度，例如：

[[[1] 
  [2]]
 [[3]
  [4]]]

有一种简单的方法吗？

Answer 1

使用np.newaxis和tolist：

a = np.array([[1, 2], [3, 4]])
a[:, :, np.newaxis].tolist()

输出：

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