辞書の配列から二次元配列を生成したい

Question

辞書の配列から二次元配列を生成したい。

data=[{'A':'A','S':'apple','W':'NY'},{'A':'B','S':'windows','W':'CF'},{'A':'B','S':'Lenovo','W':'CH'},{'A':'A','S':'summung','W':'KL'}]

という配列がある。
この配列を

ans=[[['A','A'],['S','apple'],['W','NY']],[['A','B'],['S','windows'],['W','CF']],[['A','B'],['S','Lenovo'],['W','CH']],[['A','A'],['S','summung'],['W','KL']]]

という二次元配列にしたい。

for d in range(len(data)):
    ky = data[d].keys()
    vl = data[d].values()
    total = []
    ans=[]
    for i in range(len(ky)):
        itm=[ky[i],vl[i]]
        total.append(itm)
    ans.append(total)

とコードを書いた。このコードを実行すると、

    itm=[ky[i],vl[i]]
TypeError: 'dict_keys' object does not support indexing

とエラーが出た。

for文を使いながら辞書のキーを指定する方法がわからなかった。
どのようにコードを修正すればいいか？

Answer 1

data=[{'A':'A','S':'apple','W':'NY'},{'A':'B','S':'windows','W':'CF'},
      {'A':'B','S':'Lenovo','W':'CH'},{'A':'A','S':'summung','W':'KL'}]

result = []
for dict_ele in data:
    result.append([])
    for key, value in dict_ele.items():
        result[-1].append([key, value])
print(result)

itemsを使ってforをすればkeyとvalueを取り出せるので、やりやすいです。

順序合わせたい場合

result = []
for dict_ele in data:
    result.append([])
    for key in ["A", "S", "W"]:
        result[-1].append([key, dict_ele[key]])
print(result)

Answer 2

順番が不定でよいなら、一応こんな方法も有ります。

ans = [list(map(list, d.items())) for d in data]

Answer 3

キーが事前にわかっているなら、内包表記1行で書けます。

from pprint import pprint

data=[{'A':'A','S':'apple','W':'NY'},{'A':'B','S':'windows','W':'CF'},{'A':'B','S':'Lenovo','W':'CH'},{'A':'A','S':'summung','W':'KL'}]

ans = [[[k,d[k]] for k in ["A", "S", "W"]] for d in data]
pprint(ans)
""" =>
[[['A', 'A'], ['S', 'apple'], ['W', 'NY']],
 [['A', 'B'], ['S', 'windows'], ['W', 'CF']],
 [['A', 'B'], ['S', 'Lenovo'], ['W', 'CH']],
 [['A', 'A'], ['S', 'summung'], ['W', 'KL']]]
"""

---

これは3次元のリストです（敢えて言えば）。
ネストを深くしてもアクセスに余計な手間がかかるだけなので、最初から0番目の要素は"A"キーに対応・・・という風に決めてしまった方がラクでしょう。

ans = [[d[k] for k in ["A", "S", "W"]] for d in data]
pprint(ans)

""" =>
[['A', 'apple', 'NY'],
 ['B', 'windows', 'CF'],
 ['B', 'Lenovo', 'CH'],
 ['A', 'summung', 'KL']]
"""

ans[0][0]でアクセスできてラクですが、冷静に考えたら元のもdata[0]["A"]でアクセスできるので、何が良いのかよくわからないことになりました。何がしたかったんでしたっけ。

Answer 4

items()メソッドは、辞書のキーと値のタプルのリストを作るので、そちらを使った方がよい。
なお、イテラブルなオブジェクトをインデクシングするには、list()でリスト化するとよい。