辞書型のデータの2次元配列への格納方法

Question

before_sorted=[ [{"lang": "en", "value": "CWE-20"}, {"lang": "en", "value": "CWE-90"}],
[{"lang": "en", "value": "CWE-200"}, {"lang": "en", "value": "CWE-190"},
{"lang:"en", "value": "CWE-220"}]]

上のデータはJSONのデータを辞書型に変換したものですが、上記データから下記のような出力を得たいです。

final_result_sorted = [["CWE-20", "CWE-90"], ["CWE-200", "CWE-190", "CWE-200"]]

result_sorted1 = []
result_sorted2 = []
final_result = []


for i in range(len (before_sorted)):
    for k in range(len(before_sorted[i])):
        if i == 0:
            result_sorted_1.append(before_sorted[i][k]["value"])

        elif i==1:
 result_sorted_2.append(before_sorted[i][k]["value"])

final_result=[result_sorted_1, result_sorted_2]

print(final_result)

とすれば、期待出力自体は得られるのですが、配列の数の分、変数が増えてしまいます。

効率の良いコードの書き方をご存知でしたら、ご教示いただけないでしょうか。スマホ記入のため、インデントがおかしいところがあると思いますが、後で修正します。

Answer 1

リスト内包表記を使ってください。

>>> before_sorted=[ [{"lang": "en", "value": "CWE-20"}, {"lang": "en", "value": "CWE-90"}],
[{"lang": "en", "value": "CWE-200"}, {"lang": "en", "value": "CWE-190"},
{"lang":"en", "value": "CWE-220"}]]
>>> [[d["value"] for d in l] for l in before_sorted]
[['CWE-20', 'CWE-90'], ['CWE-200', 'CWE-190', 'CWE-220']]

通常のループでやる場合でも、別に配列の数だけ変数を作る必要はないんじゃないでしょうか。

>>> result = []
>>> for l in before_sorted:
...     tmp = []
...     for d in l:
...         tmp.append(d["value"])
...     result.append(tmp)
... 
>>> result
[['CWE-20', 'CWE-90'], ['CWE-200', 'CWE-190', 'CWE-220']]

Comments

[退会済みユーザー]

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

Answer 2

効率を突き詰めると以下のようになります。

from operator import itemgetter

final_result = [[*map(itemgetter("value"), chunk)] for chunk in before_sorted]

「もうちょっと読めるように」と考えると以下の様にコードを組んでいったら良いと思います。

final_result = []
for chunk in before_sorted:
    values = []
    for item in chunk:
        # valuesが["CWE-20", "CWE-90"]みたいになるように組み立てる
    final_result.append(values)

Comments

[hayataka2049]

少なくとも速度に関しては素直な内包表記の方が速くありませんか？

[YouheiSakurai]

速度に関してはそうだと思います、関数コールのコストも馬鹿になりませんしね。でもなんとなく（Flat is better than nested.的に）2重の内包表記は読みづらくて、あと込み入ったときにPEP8準拠が難しい気がして私は避ける傾向にあります。まぁ完全に個人の趣味嗜好の範囲ですが。

[退会済みユーザー]

色々な書き方を教えて下さり、ありがとうございました。