Python - 将列表转换为列表列表

时间:2016-08-14 03:48:50

标签: python list nested-lists

我正在尝试创建一个列表列表,这样每个内部列表都有8个元素,在python单行中。

到目前为止,我有以下内容:

locations = [[alphabet.index(j) for j in test]]

映射到列表中的一个大列表:

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

但我想把它分成多个内部列表,每个8个元素:

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

知道如何实现这个目标吗?

2 个答案:

答案 0 :(得分:1)

使用range()列表切片来获取起始索引:

In [3]: test = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]

In [4]: [test[i:i+8] for i in range(0, len(test), 8)]
Out[4]: [[1, 2, 3, 4, 5, 6, 7, 8], [9, 10, 11, 12, 13, 14, 15, 16]]

作为一项功能:

In [7]: def slicing(list_, elem_):
   ...:     return [list_[i:i+elem_] for i in range(0, len(list_), elem_)]

In [8]: slicing(test, 8)
Out[8]: [[1, 2, 3, 4, 5, 6, 7, 8], [9, 10, 11, 12, 13, 14, 15, 16]]

答案 1 :(得分:0)

另一种解决方案可能是使用NumPy

import numpy as np

data = [x for x in xrange(0, 64)]
data_split = np.array_split(np.asarray(data), 8)

输出:

for a in data_split:
    print a

[0 1 2 3 4 5 6 7]
[ 8  9 10 11 12 13 14 15]
[16 17 18 19 20 21 22 23]
[24 25 26 27 28 29 30 31]
[32 33 34 35 36 37 38 39]
[40 41 42 43 44 45 46 47]
[48 49 50 51 52 53 54 55]
[56 57 58 59 60 61 62 63]