Python:リストの連結・操作がわからない

Question

###リストの連結・操作がわからない
Pythonを使用しています。
「a」のリストを「b」のリストの条件で、「c」のリストを出力したいのですが、
方法がわからないです。

#元のリスト
a = [[0,0], [1,1], [2,2], [3,3], [4,4], [5,5], [6,6], [7,7], [8,8], [9,9]]

#条件のリスト
b = [2,4,1,1,2]

#実際に出力したいリスト
c = [[[0,0], [1,1]], [[2,2], [3,3], [4,4], [5,5]], [[6,6]], [[7,7]], [[8,8], [9,9]]]

Answer 1

簡易に書くならこんな感じでしょうか。
少なからず丸投げな感があるので詳細な解説は避けます。

c = []
idx = 0
for length in b:
    tmp = []
    for _ in range(length):
        tmp.append(a[idx])
        idx += 1
        
    c.append(tmp)
        
print(c)

---

私ならmore-itertoolsのsplit_intoを使います。

c = list(
    more_itertools.split_into(a, b)
)

Comments

[TeRa_130]

ありがとうございます。

Answer 2

b は各 list の数をもってます。
これを a の部分 list a[x:y] の形式でとりだせるように、 b を変形します。

[sum(b[0:x]) for x in range(len(b))] + [sum(b)]

とすれば、[2,4,1,1,2] が [0, 2, 6, 7, 8, 10] になります。
この list は、求めたい a の部分リストの先頭位置を持っています。(最後の 10 をのぞいて)