数字の要素をキーにして配列から辞書を作りたい

Question

前提・実現したいこと

数字の要素をキーにして配列から辞書を作りたい。
今配列listsに

lists = ["1","東京","神奈川","千葉","2","群馬","茨城","山梨","3","長野","栃木","4","愛知"]

のような要素が入っている。
このlistsを

dict = {1:("東京","神奈川","千葉"),2:("群馬","茨城","山梨"),3:("長野","栃木"),4:("愛知")}

のような辞書に変換したい。

該当のソースコード

　　 dict = {}
    for list in lists:
        if type(list) == int:
            dict[list]
        else:
            dict = list
    print(dict)

のように書いて実行した。

試したこと

該当のソースコードを実行したこと。

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

該当のソースコードでは目的のアウトプットが得られない。

Comments

[wakame]

リスト内文字列であれば""か''でくくってもらえませんか。これソースコードからのコピペなら変数未定義でエラーでそうですけど。

[hayataka2049]

とりあえず、listという変数名は組み込みのlistクラスを上書きしちゃうので、使ってはいけませんとアドバイスしておきます

Answer 1

ちょっと泥臭いですが、こんな感じで書けました。

lists = [1, '東京', '神奈川', '千葉', 2, '群馬', '茨城', '山梨', 3, '長野', '栃木', 4, '愛知']

dst = {}
for elem in lists:
    if isinstance(elem, int):
        dst[elem] = []
        pre_key = elem
    else:
        dst[pre_key].append(elem)
        
print(dst)

実行結果 Wandbox

{1: ['東京', '神奈川', '千葉'], 2: ['群馬', '茨城', '山梨'], 3: ['長野', '栃木'], 4: ['愛知']}

追記

もしキーが『数値』ではなく『数字』なら。Wandbox

lists = ['1', '東京', '神奈川', '千葉', '2', '群馬', '茨城', '山梨', '3', '長野', '栃木', '4', '愛知']

dst = {}
for elem in lists:
    try:
        elem = int(elem)
        dst[elem] = []
        pre_key = elem
    except ValueError:
        dst[pre_key].append(elem)

print(dst)

Comments

[wakame]

Wandbox、コードと実行結果、かつそれらを動かす環境を提示できるのですねすごい。

[LouiS0616]

いつの間にか自分の中での定型がこうなってたのです。
動作チェックも忘れずにすみますし。

[退会済みユーザー]

回答をくださりありがとうございます。提示してくださったコードを実行したのですが、UnboundLocalError: local variable 'pre_key' referenced before assignment　とエラーが出てしまいます。else文の中のdst[pre_key].append(elem)　のpre_keyにアクセスができないようです。pre_keyの初期値は何に設定しておけば良いでしょうか？

[LouiS0616]

listsの最初の要素は数値ではないのですか？
あるいは、数値ではなく『数字』なのでしょうか。

[退会済みユーザー]

listsの最初の要素は文字列になっていました！この場合だとコードに影響がありますか？

[LouiS0616]

文字列というと、数値を表す文字列ですか？地名を表す文字列ですか？
前者の場合は追記したコードで動作します。
後者の場合は、minmiさんがどのように扱いたいかに依ります。

[退会済みユーザー]

はい、数値を表す文字列です。目的のアウトプットが出ました！ありがとうございます。また、intとstrを辞書のキーにするときの扱いの違いもわかり勉強になりました。

[hayataka2049]

敢えてコードにケチ付けると、isnumericでも使って判定すれば・・・何もtry-exceptなんかしなくても・・・＞LouiS0616さん

[LouiS0616]

@hayataka2049 さん
確かに自然数しか扱わない場合はisnumericで充分ですね。ご指摘ありがとうございます。

Answer 2

わかりにくさを追求してみました。
https://wandbox.org/permlink/0WKHUDKbRjZAghCL

import re

lists = ['1', '東京', '神奈川', '千葉', '2', '群馬', '茨城', '山梨', '3', '長野', '栃木', '4', '愛知']

def judge(L):
    k = L[0]
    vs = []
    for v in L[1:]:
        if re.fullmatch(r'\d*', v):
            yield (k, vs)
            k = v
            vs = []
        else:
            vs.append(v)
    yield (k, vs)

dst = dict(judge(lists))

print(dst)

Answer 3

2 つの方法を書いてみました,

lists = [1,"東京","神奈川","千葉",2,"群馬","茨城","山梨",3,"長野","栃木",4,"愛知"]

dic = {}
key = None
for elem in lists:
    if isinstance(elem, int):
        key = elem
        dic[key] = []
    else:
        dic[key].append(elem)

print(dic)


key_pos = [[x, pos + 1] for pos, x in enumerate(lists) if isinstance(x, int)]
keys = [x[0] for x in key_pos]
poss = [x[1] for x in key_pos] + [len(lists) + 1]
vals = [lists[poss[idx]:poss[idx + 1] - 1] for idx in range(len(keys))]
dic = dict(zip(keys, vals))
print(dic)

Answer 4

別解です。
数字のリストと、数字に対応する県名のリストをパッキングしました。

import re

japan_pref_infos = [
    "1","東京","神奈川","千葉","2","群馬","茨城","山梨","3","長野","栃木","4","愛知"
]

seq = ' '.join(japan_pref_infos)
japan_prefs = dict(zip(
    map(int, re.findall('\d', seq)),
    map(lambda x: tuple(x.strip().split()), re.split('\d', seq)[1:])
))

print(japan_prefs)

Answer 5

lists = [1,"東京","神奈川","千葉",2,"群馬","茨城","山梨",3,"長野","栃木",4,"愛知"]

result = {}
current_num = None
for item in lists:
    if isinstance(item, int):
        current_num = item
        result[current_num] = []
    else:
        result[current_num].append(item)

print(result)
# => {1: ['東京', '神奈川', '千葉'], 2: ['群馬', '茨城', '山梨'], 3: ['長野', '栃木'], 4: ['愛知']}

Comments

[yukkeorg]

おおっと．．．すでに回答者がお二方いらっしゃいましたね。

Answer 6

以下のコードで希望の出力になります。

>>> data = [1,"東京","神奈川","千葉",2,"群馬","茨城","山梨",3,"長野","栃木",4,"愛知"]
>>> d = dict()
>>> for elem in data:
...     if type(elem) == int:
...         last_key = elem
...         d[last_key] = list()
...     else:
...         d[last_key].append(elem)
... 
>>> d
{1: ['東京', '神奈川', '千葉'], 2: ['群馬', '茨城', '山梨'], 3: ['長野', '栃木'], 4: ['愛知']}