python:根据指标列表将一个值列表划分为一组列表

Question

例如。

list = [[245, 187], [69, 32] ,
[892, 176], [133, 120],
[161, 26], [62, 95]]
conditionslist2 = [0,1,2,0,0,2]

期望结果

list0 = [[ 245  , 187],[133 , 120] ,   [161 , 26]]
list1 = [[ 69 ,  32]]
list2 = [[ 892 , 176],[ 62 , 95]]

Answer 1

IIUC，您希望根据输出中提供的索引列表拆分输入。

不要生成变量名，使用字典：

lst = [[245, 187], [69, 32] ,
[892, 176], [133, 120],
[161, 26], [62, 95]]
conditionslist2 = [0,1,2,0,0,2]
from collections import defaultdict
out = defaultdict(list)
for i, sublist in zip(conditionslist2, lst):
out[f'list{i}'].append(sublist)

dict(out)

输出：

{'list0': [[245, 187], [133, 120], [161, 26]],
'list1': [[69, 32]],
'list2': [[892, 176], [62, 95]]}