二つの配列を使ってデータを整理したい

前提・実現したいこと

要素数が同じ二つの配列(list, value)を使って
list = [1,0,0,1,2]
value = [[1,2,3],[4,5,6],[7,8,9],[10,11,12],[13,14,15]]

listの数字ごとにvalueを以下のように纏めたいのですが、何か方法はありますでしょうか。
list0 = [[4,5,6],[7,8,9]]
list1 = [[1,2,3],[10,11,12]]
list2 = [[13,14,15]]

meg_

2020/08/25 18:24

> listの数字ごとにどういう意味でしょうか？

jeanbiego

2020/08/25 23:34

「何か方法はありますでしょうか」だけだと、「あります」とだけ答えたくなります。ご自分で書かれたコードを記載して、その改善・修正を依頼するのが良いですよ。 https://teratail.com/help/question-tips

行動規範の内容に同意します

回答2件

Python
1lst = [1, 0, 0, 1, 2]
2values = [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12], [13, 14, 15]]
3
4for n in set(lst):
5    exec(f'list{n} = {[x for i, x in enumerate(values) if lst[i] == n]}')
6
7print(list0)
8print(list1)
9print(list2)

terminal
1[[4, 5, 6], [7, 8, 9]]
2[[1, 2, 3], [10, 11, 12]]
3[[13, 14, 15]]

投稿2020/08/26 03:13

Daregada

総合スコア11990

Daregada

2020/08/26 03:16

あたりまえだが、VSCodeに「Undefined variable: 'list0'」などと警告を出される。

行動規範の内容に同意します

ベストアンサー

やりようはいくつもありますが、例えば結果をdictにしてよければdefaultdictを利用して次のように書けるかもしれません。dictの要素の並びは基本的に不定になるので、必要に応じてlistやOrderdDictを使ってソートしてください。

Python
1from collections import defaultdict
2lst = [1,0,0,1,2]
3values = [[1,2,3],[4,5,6],[7,8,9],[10,11,12],[13,14,15]]
4
5d = defaultdict(list)
6for (idx, value) in zip(lst, values):
7    d[idx].append(value)
8
9
10# 結果の確認用
11from pprint import pprint
12pprint(d)
13# defaultdict(<class 'list'>,
14#             {0: [[4, 5, 6], [7, 8, 9]],
15#              1: [[1, 2, 3], [10, 11, 12]],
16#              2: [[13, 14, 15]]})