python - 如何创建嵌套列表并在每个嵌套中放置特定数量的元素？列表和它的嵌套元素总是不一样的

Question

这就是我正在做的，

    amount = [2, 2, 3]
my_list = ['name_zero', 'name_one', 'score_three', 'score_nine', 'age_zero', 'age_ten', 'age_six']
output = []

for i in amount:
    output.append(my_list[:i])

print(output)

• 我得到了什么:

  [['name_zero', 'name_one'], ['name_zero', 'name_one'], ['name_zero', 'name_one', 'score_three']]

我的代码的问题是，它附加了相同的元素，接收这样的输出的正确方法应该是什么？

# [['name_zero', 'name_one'], ['score_three', 'score_nine'], ['age_zero', 'age_ten', 'age_six']]

Answer 1

尝试:

amount = [2, 2, 3]
my_list = [
    "name_zero",
    "name_one",
    "score_three",
    "score_nine",
    "age_zero",
    "age_ten",
    "age_six",
]
output = []

i = iter(my_list)
for v in amount:
    output.append([next(i) for _ in range(v)])

print(output)

打印:

[
    ["name_zero", "name_one"],
    ["score_three", "score_nine"],
    ["age_zero", "age_ten", "age_six"],
]