<python3>辞書型配列のvalueの操作についての質問

Question

前提・実現したいこと

#現状の辞書型配列
↓
dict = {0: ['s', 'b', 'd', 't'], 1: ['s', 'a', 'c', 't'], 2: ['s', 'b', 'c'], 3: ['s', 'a', 'c']}

#最終的に表現したい辞書型配列
↓
new_dict = {0: [['s', 'b'], ['b', 'd'], ['d', 't']], 1: [['s', 'a'], ['a', 'c'], ['c', 't']], 2: [['s', 'b'], ['b', 'c']], 3: [['s', 'a'], ['a', 'c']]}

上記のようにdictという辞書型配列のvalueに関して、それぞれ連続する右隣の文字列とでリストを形成し、それぞれのkeyのvalueを二重配列に改変したいです。

その方法についてご教示頂けたらと存じます！

#現状の辞書型配列
dict = {0: ['s', 'b', 'd', 't'], 1: ['s', 'a', 'c', 't'], 2: ['s', 'b', 'c'], 3: ['s', 'a', 'c']}


#実現したい辞書型配列
new_dict = {0: [['s', 'b'], ['b', 'd'], ['d', 't']], 1: [['s', 'a'], ['a', 'c'], ['c', 't']], 2: [['s', 'b'], ['b', 'c']], 3: [['s', 'a'], ['a', 'c']]}

試したこと

[['s', 'b'],
['b', 'd'],
['d', 't'],
['s', 'a'],
['a', 'c'],
['c', 't'],
['s', 'b'],
['b', 'c'],
['s', 'a'],
['a', 'c']]
このようにペアを作り、
⑴分割し、valueごとのリストを作成し
⑵keyとzipする
と手順は考えていたのですが、まず⑴でつまずいてしまいました。

Answer 1

参考までに、痒いところに手が届くRubyの例です。

src = {0 => ['s', 'b', 'd', 't'], 1 => ['s', 'a', 'c', 't'], 
2 => ['s', 'b', 'c'], 3 => ['s', 'a', 'c']}

dst = src.transform_values{|x| x.each_cons(2).to_a}

quiquiさんの紹介されているpairwise相当がeach_cons(2)です。

Comments

[FoiFoi]

素晴らしくスマートに表現することができるんですね、共有頂きありがとうございます！

[quickquip]

高速に動作する必要があるものだけitertoolsで用意しておいて、その組み合わせでどうにかなるものは自分で書け（あるいはレシピに書いておくからそれを使え）がPythonの思想なのですよね……。
レシピに書いてある基本的な作用ぐらいサブパッケージにでも入れておいてくれればいいのにとよく思います。
ちなみに each_cons(n) は Pythonだと zip(*starmap(islice, zip(tee(seq, n), count(0), repeat(None)))) が簡潔(?)かつ最速だと思います。

Answer 2

Python公式ドキュメントのitertoolosのレシピに、pairwiseという関数が例として書いてあります。

>>> from itertools import tee

>>> def pairwise(iterable):
...     "s -> (s0,s1), (s1,s2), (s2, s3), ..."
...     a, b = tee(iterable)
...     next(b, None)
...     return zip(a, b)

と準備したら、（関数はitertoolosのレシピからの引用です）

>>> d =  {0: ['s', 'b', 'd', 't'], 1: ['s', 'a', 'c', 't'], 2: ['s', 'b', 'c'], 3: ['s', 'a', 'c']}

>>> new_dict = {k: list(pairwise(v)) for k, v in d.items()}

>>> new_dict
{0: [('s', 'b'), ('b', 'd'), ('d', 't')], 1: [('s', 'a'), ('a', 'c'), ('c', 't')], 2: [('s', 'b'), ('b', 'c')], 3: [('s', 'a'), ('a', 'c')]}

でほぼ欲しいものになってると思います。(変数名dictは標準ライブラリの辞書の型名を隠してしまうのでやめましょう）

タプルなのがどうしても嫌なら、

>>> new_dict = {k: list(map(list, pairwise(v))) for k, v in d.items()}

>>> new_dict
{0: [['s', 'b'], ['b', 'd'], ['d', 't']], 1: [['s', 'a'], ['a', 'c'], ['c', 't']], 2: [['s', 'b'], ['b', 'c']], 3: [['s', 'a'], ['a', 'c']]}

で。

Comments

[FoiFoi]

素晴らしい回答ありがとうございます。

ただ今回私が理想としていたのが、モジュールを使わない形での表現方法だったので申し訳ございません。
しかし今後の参考にさせて頂きます！

> 変数名dictは標準ライブラリの辞書の型名を隠してしまうのでやめましょう
ありがとうございます、ローカルで定義していた変数名があまりにヘンテコだったので単純にdictと定義してしまいました。アドバイス頂きありがとうございます。

Answer 3

面白そうだったので書いてみました。

src = {0: ['s', 'b', 'd', 't'], 1: ['s', 'a', 'c', 't'], 2: ['s', 'b', 'c'], 3: ['s', 'a', 'c']}

dst = {}
for k, v in src.items():
    tmp = []
    for p in zip(v, v[1:]):
        p = list(p)
        tmp.append(p)
        
    dst[k] = tmp
        
print(dst)

実行結果 Wandbox

{0: [['s', 'b'], ['b', 'd'], ['d', 't']], 1: [['s', 'a'], ['a', 'c'], ['c', 't']], 2: [['s', 'b'], ['b', 'c']], 3: [['s', 'a'], ['a', 'c']]}

---

なお、内包表記を使うとこんなふうに書けます。Wandbox

src = {0: ['s', 'b', 'd', 't'], 1: ['s', 'a', 'c', 't'], 2: ['s', 'b', 'c'], 3: ['s', 'a', 'c']}
dst = {
    k: [list(p) for p in zip(v, v[1:])]
    for k, v in src.items()
}

print(dst)

Comments

[FoiFoi]

内包表記での表現までご教示頂きありがとうございました。
まさに内包表記までのアウトプットこそ理想としていた形でしたので、ベストアンサーとさせて頂きました。

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

[FoiFoi]

追って失礼します。
for p in zip(v, v[1:]):
上記部分について咀嚼させて頂いてましたが、
素晴らしくスマートで感動いたしました。

ありがとうございました！