pythonの配列の扱い方について質問です

Question

前提・実現したいこと

allay = [[1,a],[1,b][1,c],[2,d],[2,e],[3,f],...[1,z].]

上のような配列があった時に、下のように一つ目の値を元に、２つ目の値を分類したいです

class1 = [a,b,c,z]
class2 = [d,e]
class3 = [f]
…

発生している問題・エラーメッセージ

現在は下のように、あらかじめ class1,class2,class3のような空の配列を宣言し、if文を使って分類しています。

for i in(0,len(allay)):
  if allay[i] == 1:
    class1.append(allay[i])
  elif allay[i] == 2:
    class2.append(allay[i])

困っていることは、分類するクラス１００個など多くなった時、コードの量が増えてしまうことです。
もっと簡単にできる方法はあるのでしょうか？

よろしくお願いいたします。

Answer 1

Dictionaryをつかうといいです。

Comments

[rekna]

ご回答ありがとうございました。

Answer 2

a = [[1,'a'],[1,'b'],[1,'c'],[2,'d'],[2,'e'],[3,'f'],[1,'z']]
cls = {}
for l in a:
    if l[0] not in cls:
        cls[l[0]] = []
    cls[l[0]].append(l[1])

print(cls) # {1: ['a', 'b', 'c', 'z'], 2: ['d', 'e'], 3: ['f']}

Comments

[rekna]

解決しました。
ありがとうございました。