Python:リストを複数のリストに分割したい

Question

リストを複数の別のリストに分割したいです。

a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
b = [3, 4, 7]

リストaをリストbに含まれている値の位置で分割していきたいです。

#b[3]のとき
c = [1, 2, 3, 4]
#b[4]のとき
d = [5]
#b[7]のとき
e = [6, 7, 8]
#残り
f = [9, 10]

上記の例ですと、以上のような分け方にしたいのですが、皆目見当がつきません。
リストをn個のリストに分割するのではなく、別のリストで与えた値の位置で分割するにはどのようにすればよいのでしょうか？
リストを作るための空のリストをどのように用意しておけばいいのかという点からすでにわかりません。aもbもあらかじめ要素の数がわからない（bの要素数がaの要素数を超えることはない）ため、空のリストをあらかじめ用意することはできないように思っています。
そもそも以上のような処理は現実的なのでしょうか？
ご教示いただけますと幸いです。よろしくお願いいたします。

Answer 1

numpy.array_split() を使うとよいと思います。
第1引数のリストを第2引数で指定されたインデックスの直前で分割します。
今回はインデックスの直後で分割するとのことだったので、np.array(b) + 1 としています。

import numpy as np

a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
b = [3, 4, 7]

split_array = np.array_split(a, np.array(b) + 1)
print(split_array)  # [array([1, 2, 3, 4]), array([5]), array([6, 7, 8]), array([ 9, 10])]

Comments

[Ykkykk]

ご回答いただきありがとうございます。
numpyを使用してもできるのですね。
この場合、split_array内のリストの要素にアクセスする際は、通常のネストされたリストと同じように添え字で指定することができるのでしょうか？

[tiitoi]

通常のリストとまったく同じ使い方ができます。

print(split_array[0]) # [1 2 3 4]
print(split_array[0][1]) # 2

[Ykkykk]

なるほど！よくわかりました。
ありがとうございます！

Answer 2

a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
separate = [3, 4, 7]

start = 0
newArray = []
for s in separate :
    newArray.append(a[start:s+1])
    start = s+1
newArray.append(a[start:])

print(newArray)

変数を動的に作るのはできないので、配列の中に配列を入れています

array[a:b]というのは配列の中のa番目からb番目の直前までを抜き出すという意味です

Comments

[Ykkykk]

ご回答いただきありがとうございます。
やはり動的に変数を作ることはできないのですね…。
配列の配列にしてアクセスすることになるということですね。
ありがとうございます！

Answer 3

separate[x:y] で [separate[x], ..., separatey[y -1]] を取り出せることを利用します。

sep.py

a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
separate = [3, 4, 7]

sp = [0] + [x +1 for x in separate] + [len(a)]
result = [a[sp[i]:sp[i + 1]] for i in range(len(sp) - 1)]
print(result)

上のコードでは、 sp = [0, 4, 5, 8, 10] となります。
そして a[0:4], a[4:5], a[5:8], a[8:10] のように取り出していっています。

実行例