强制numpy将列表保留为列表

Question

x2_Kaxs是一个Nx3 numpy列表数组,这些列表中的元素索引到另一个数组.我想最终得到这些索引元素的Nx3 numpy数组列表.

x2_Kcids = array([ ax2_cid[axs] for axs in x2_Kaxs.flat ], dtype=object)

这输出一个(N*3)x1阵列的numpy数组.大.这几乎适用于我想要的东西.我需要做的就是重塑它.

x2_Kcids.shape = x2_Kaxs.shape

这很有效.x2_Kcids成为Nump3阵列的numpy数组.完善.

除了所有列表中x2_Kaxs只有一个元素.然后它将它展平为一个Nx3整数数组,我的代码需要在管道中稍后列出一个列表.

我提出的一个解决方案是添加一个虚拟元素,然后将其弹出,但这非常难看.还有什么更好的吗？

Answer 1

类似于@Denis：

if x.ndim == 2:
    x.shape += (1,)