リストを分解したいが重複してしまう

Question

リストを分解したいです。

[['hello','my','job',['aaa','bbb']],['we','i',['test','item']],['she','is','my','bos',['test','item']]]

というリストがあります。このリストを、

[['hello','my','job'],['we','i'],['she','is','my','bos']]

というリストと、

[['aaa','bbb'],['test','item'],['test','item']]

というリストに分けたいです。

a = []
b = []
for item in data:
    num = len(item) - 1
    for i in range(len(itemr)):
        if i != num:
             a.append(item[i])
        else:
             b.append(item[i])

とコードを書いたのですが、aリストの方が無駄に重複してしまっています。
どう修正すればいいでしょうか？

Answer 1

『無駄に重複』という表現が良く分かりませんが、
とりあえず次のように仮のリストを挟んでおけばご希望の結果にはなります。

data = [['hello','my','job',['aaa','bbb']],['we','i',['test','item']],['she','is','my','bos',['test','item']]]

a = []
b = []
for item in data:
    num = len(item) - 1
    tmp_a = []
    
    for i in range(len(item)):
        if i != num:
             tmp_a.append(item[i])
        else:
             b.append(item[i])
                
    a.append(tmp_a)
                
print(a)
print(b)

実行結果 Wandbox

[['hello', 'my', 'job'], ['we', 'i'], ['she', 'is', 'my', 'bos']]
[['aaa', 'bbb'], ['test', 'item'], ['test', 'item']]

書いてみた

要素を二回参照する無駄はありますが、この方が分かりやすいように思います。

a = [row[:-1] for row in data]
b = [row[-1] for row in data]

Wandbox

どうしても一周で完成させたい場合は、こんな感じです。

a = []
b = []

for *elems, last in data:
    a.append(elems)
    b.append(last)

Wandbox

Answer 2

絶対にお勧めしませんが、こんな書き方もできるということだけ。

data = [['hello','my','job',['aaa','bbb']],['we','i',['test','item']],['she','is','my','bos',['test','item']]]

a = []
b = []

[(a.append(d[:-1]), b.append(d[-1])) for d in data]

print(a)
print(b)

---

ちなみに、これはどうしても、lambda式内で複数行を実行したい時のテクニックです。
def使え感がありますが、書き捨てコード用。

https://stackoverflow.com/questions/862412/is-it-possible-to-have-multiple-statements-in-a-python-lambda-expression

Answer 3

2 段目の list の要素を list と そうでないものに振り分けるということで考えてみました。
2つ半の方法でかいてみました。

sep.py

from itertools import chain

data = [
    ['hello', 'my', 'job', ['aaa', 'bbb']],
    ['we', 'i', ['test', 'item']],
    ['she', 'is', 'my', 'bos', ['test','item']]
]

data_elem = [[e for e in v if not isinstance(e, list)] for v in data]
data_list = [[e for e in v if isinstance(e, list)] for v in data]
print(data_elem)
print(list(chain.from_iterable(data_list)))

print()
data_0 = list(chain.from_iterable(data))
print('data_0=', data_0)
data_list = [x for x in data_0 if isinstance(x, list)]
print(data_list)

print()
data_elem = []
data_list = []
for v in data:
    v0 = []
    for e in v:
        if isinstance(e, list):
            data_list.append(e)
        else:
            v0.append(e)
    data_elem.append(v0)

print(data_elem)
print(data_list)

実行結果