我正在尝试创建一个列表列表,这样每个内部列表都有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]]
知道如何实现这个目标吗?
答案 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]