dictデータの文字列を整数に変換したい

Question

dictデータの文字列を整数に変換したい

csvファイルを読み込み、dictデータとして計算にしたいのですが、
読み込み時に数字が文字列として認識されてしまうので、keyとvalueどちらも整数に変換したいです。

#csv取り込みからdictデータの作成
a_csv_file = open("file_name.csv", "r")
dict_reader = csv.DictReader(a_csv_file)
ordered_dict_from_csv = list(dict_reader)[0]
dict_data = dict(ordered_dict_from_csv)

#いまのdictデータ　　　　　　
dict_data = {'1': '0','2': '1','3': '1'} 

#つくりたいdictデータ
dict_data_goal = {1: 0,2: 1,3: 1}

#いまのdictデータの整数値への変換
for k in dict_data.keys():
    dict_data[k] = int(dict_data[k])
    for v in dict_2.values():
        dict_data[v] = int(dict_data[v])
print(dict_data)

以下のようなエラーメッセージが返ってきます

KeyError                                  Traceback (most recent call last)
<ipython-input-27-24eb3860fd9b> in <module>
      3     dict_data[k] = int(dict_data[k])
      4     for v in dict_2.values():
----> 5         dict_data[v] = int(dict_data[v])
      6 print(dict_data)

KeyError: 0

valueの数字は整数に変換できるのですが、keyの数字は変換できません。

環境：windows10
jupyter NotebookでPython3を使用しています。

Answer 1

dict_2.values()に対するループになっているので、dict_data[v]のキーが存在しないというエラーです。

またdict_dataに追加すると、元の文字列のものも残るので、やりたいことと違うのではないでしょうか。

シンプルに書くとこうです。

new_dict = {}
for k in dict_data.keys():
    new_dict[int(k)] = int(dict_data[k])

内包とitemsを使えば、こう書けます。

new_dict = {int(k): int(v) for k, v in dict_data.items()}

Comments

[ykar20]

ありがとうございます！教えていただいたコードで目的の変換をすることができました。