2D python列表的numpy数组到2D numpy数组

Question

这是转换类似内容的最高效方式

problem = [ [np.array([1,2,3]), np.array([4,5])],
            [np.array([6,7,8]), np.array([9,10])]]

成

desired = np.array([[1,2,3,4,5], 
                   [6,7,8,9,10]])

不幸的是,最终的列数和行数(以及子数组的长度)事先是未知的,因为从二进制文件中读取子数据,逐个记录.

Answer 1

这个怎么样:

problem = [[np.array([1,2,3]), np.array([4,5])],
        [np.array([6,7,8]), np.array([9,10])]]

print np.array([np.concatenate(x) for x in problem])