Pythonでのリスト操作 ｜ リストを分割したい

Question

###実現したいこと
Pythonでのリスト操作について質問です

[1,2,3,4,5,1,2,3,4,1,2,1,2,3,4,5,6]

のようなリストを1から始まる要素で分割して

[[1,2,3,4,5][1,2,3,4][1,2][1,2,3,4,5,6]]

という形にしたいのですが、どうすればいいのかわかりません

分割数、要素の最大値はどちらも毎回変動しますが、1,2,3,4,…というように必ず1から始まる数列です

どんな方法でも構いませんので、わかる方教えていただけませんか？

Answer 1

1のインデックスのリストを取得し、リスト内の隣り合うインデックスで元のリストをスライスすればできます。

def splitAtOnes(l):
    ones = [i for i, v in enumerate(l) if v == 1]
    ones.append(None)
    return [l[ones[i]:ones[i + 1]] for i in range(len(ones) - 1)]

splitAtOnes([1,2,3,4,5,1,2,3,4,1,2,1,2,3,4,5,6])
# -> [[1, 2, 3, 4, 5], [1, 2, 3, 4], [1, 2], [1, 2, 3, 4, 5]]

Comments

[can110]

最後、末尾まで取りたいので`append(-1)`→`append(None)`とすべきかと思います。

[karamarimo]

あ、そうですね。うっかりしてました。ありがとうございます。

Answer 2

一応こんな風に出来ますが、ちょっと美しさに欠ける印象です。
もっといい方法があるような気もします。

hoge = [1, 2, 3, 4, 5, 1, 2, 3, 4, 1, 2, 1, 2, 3, 4, 5, 6]
indices = [i for i, x in enumerate(hoge) if x == 1] + [len(hoge)]
result = [hoge[begin:end] for begin, end in zip(indices, indices[1:])]
print(result)

---

ちょっと改造。ほぼ変わりませんが。

def get_indices(arg, num):
    for i, v in enumerate(arg):
        if v == num:
            yield i

hoge = [1, 2, 3, 4, 5, 1, 2, 3, 4, 1, 2, 1, 2, 3, 4, 5, 6]
indices = list(get_indices(hoge+[1], 1))
result = [hoge[begin:end] for begin, end in zip(indices, indices[1:])]
print(result)

Answer 3

愚直に全要素をループさせて配列を構築するのがよさそうに感じます。

---

追記:

解決済みとなったようですが、実際のコードを書いておくと下記のようになるでしょうか。

arr_in = [1,2,3,4,5,1,2,3,4,1,2,1,2,3,4,5,6]
print(arr_in)

arr_out = []
arr = []
for i in arr_in:
   if i == 1:
      arr = [i]
      arr_out.append(arr)
   else:
      arr.append(i)

print(arr_out)

Comments

[sera]

わざわざコードを書いてくださったんですね
ありがとうございます

いちばんわかりやすいです

Answer 4

lst = [1,2,3,4,5,1,2,3,4,1,2,1,2,3,4,5,6]+[1]
num = [i for i,n in enumerate(lst+[1]) if n==1 or lst[i+1]==1]
num_lst = [num[i:i+2] for i in range(0,len(num),2)]
ans = [lst[sf[0]:sf[1]+1] for i,sf in enumerate(num_lst) if i!=len(num_lst)-1]
print(ans)

出力結果
[[1, 2, 3, 4, 5], [1, 2, 3, 4], [1, 2], [1, 2, 3, 4, 5, 6]]

Comments

[sera]

回答ありがとうございます

もう１つだけ質問させてください
出力結果を
[[1,2,3,4,5],[1,2,3,4],[1,2],[1,2,3,4,5,6]]
とすることはできますか？

[_Victorique__]

コードに[]を追加するだけですね。
忘れていたので追加しておきました

[sera]

すみません、少し違います...
新しく回答いただいた方のように [ ]の中がリストになってほしいんです

[_Victorique__]

すみません、修正したのでご確認ください

[sera]

何度もすみませんでした
ありがとうございます