获取 python 列表中第一个元素的索引

Question

我有一个这样的列表：

[[21], [22, 1], [23, 2], [24, 1, 0, 4], [25, 2, 3]]

我想 append 到一个新列表中每个子列表的第一项的索引，同时计算所有子列表中的所有元素。 新列表的每个索引取决于每个子列表的长度。 我正在寻找的结果是这个：

[0、1、3、5、9]。

我不想使用 numpy 或任何其他库。 谢谢！

Answer 1

累加从零开始的长度：

L = [[21], [22, 1], [23, 2], [24, 1, 0, 4], [25, 2, 3]]

from itertools import accumulate
idx = [0,*accumulate(map(len,L[:-1]))]

print(idx) # [0, 1, 3, 5, 9]

Answer 2

你需要这样的东西：

l = [[21], [22, 1], [23, 2], [24, 1, 0, 4], [25, 2, 3]]

n = []
p = 0
for s1 in l:
    n.append(p)
    p += len(s1)

print(n)