如何在python中将平面列表转换为2D数组

Question

如何打开诸如之类的列表

data_list = [0,1,2,3,4,5,6,7,8,9]

转换成一个数组(我使用的是numpy(，看起来像：

data_array = [ [0,1] , [2,3] , [4,5] , [6,7] , [8,9] ]

我可以从列表的开头切片并将它们附加到一个空数组中吗？

感谢

Answer 1

>>> import numpy as np
>>> np.array(data_list).reshape(-1, 2)
array([[0, 1],
       [2, 3],
       [4, 5],
       [6, 7],
       [8, 9]])

(reshape方法在数组上返回一个新的"视图"；它不复制数据。(

Answer 2

def nest_list(list1,rows, columns):    
        result=[]               
        start = 0
        end = columns
        for i in range(rows): 
            result.append(list1[start:end])
            start +=columns
            end += columns
        return result

用于：

 list1=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
nest_list(list1,4,4)

输出：

[[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11], [12, 13, 14, 15]]