python - 在 python 中用循环填充数组

Question

我的问题如下，我有一个名为 crave=['word1','word2','word3'....'wordx'] 的数组，我想转换成 ITEM=[['word1','word2','word3'],['word4','word5','word6'] 等]我使用了以下代码

buff=[['none','none','none']]
n=10
ITEM=[]
i=0
while 1>0 :


 buff[0][0]=crave[i]
 buff[0][1]=crave[i+1]
 buff[0][2]=crave[i+2]
 ITEM.insert(len(ITEM),buff[0])
 i=i+3
 if i>n:
      break

但我得到的是 [['wordx-2','wordx-1','wordx'],['wordx-2','wordx-1','wordx'],[ 'wordx-2','wordx-1','wordx']等]为什么会这样？:(

Answer 1

您可以使用列表理解轻松地做到这一点。xrange(0, len(crave), 3) 用于从 0 迭代到 len(crave)，间隔为 3。然后我们使用 list slicing crave[i:i+3] 提取所需数据。

crave=['word1','word2','word3','word4','word5','word6','word7']

ITEM = [crave[i:i+3] for i in xrange(0, len(crave), 3)]

print ITEM
>>> [['word1', 'word2', 'word3'], ['word4', 'word5', 'word6'], ['word7']]