Как создать вложенный dict в Python?
у меня есть 2 csv-файла. Первый файл данных и файл сопоставления. Файл сопоставления имеет 4 столбца Device_Name GDN Device_Type Device_OS
Это также столбцы, которые присутствуют в файле данных и должны быть обработаны.
файл данных содержит данные с заполненным столбцом Device_Name и пустыми столбцами rest 3. Файл сопоставления содержит все заполненные столбцы. Я хочу, чтобы мой код Python открывал оба файла и для каждого имени устройства в файле данных отображал его значение GDN, Device_Type & Device_OS из сопоставления файл.
Я знаю, как использовать dict, когда присутствуют только 2 столбца (1 необходимо сопоставить), но я не знаю, как это сделать, когда 3 столбца должны быть сопоставлены.
Ниже приведен код, с помощью которого я пытался выполнить отображение Device_Type
:
x = dict([])
with open("Pricing Mapping_2013-04-22.csv", "rb") as in_file1:
file_map = csv.reader(in_file1, delimiter=',')
for row in file_map:
typemap = [row[0],row[2]]
x.append(typemap)
with open("Pricing_Updated_Cleaned.csv", "rb") as in_file2, open("Data Scraper_GDN.csv", "wb") as out_file:
writer = csv.writer(out_file, delimiter=',')
for row in csv.reader(in_file2, delimiter=','):
try:
row[27] = x[row[11]]
except KeyError:
row[27] = ""
writer.writerow(row)
возвращает Atribute Error
.
после некоторых исследований я понял, что мне нужно создать вложенный dict, но я понятия не имею, как это сделать. Пожалуйста, помогите мне в решении этого или подтолкните меня в правильном направлении, чтобы решить эту проблему.
4 ответов
вложенный dict-это словарь в словаре. Очень простая вещь.
>>> d = {}
>>> d['dict1'] = {}
>>> d['dict1']['innerkey'] = 'value'
>>> d
{'dict1': {'innerkey': 'value'}}
вы также можете использовать defaultdict
С collections
пакет для облегчения создания вложенных словарей.
>>> import collections
>>> d = collections.defaultdict(dict)
>>> d['dict1']['innerkey'] = 'value'
>>> d # currently a defaultdict type
defaultdict(<type 'dict'>, {'dict1': {'innerkey': 'value'}})
>>> dict(d) # but is exactly like a normal dictionary.
{'dict1': {'innerkey': 'value'}}
вы можете заполнить это, как хотите.
Я бы рекомендовал в вашем коде что-то как следующее:
d = {} # can use defaultdict(dict) instead
for row in file_map:
# derive row key from something
# when using defaultdict, we can skip the next step creating a dictionary on row_key
d[row_key] = {}
for idx, col in enumerate(row):
d[row_key][idx] = col
по вашему комментарий:
может быть выше код запутанный вопрос. Моя проблема в двух словах: я есть 2 файла a.csv b.csv, a.csv имеет 4 столбца i j k l, b.csv также имеет этот столбец. i является своего рода ключевыми столбцами для этих CSV. столбец j k л пусто в a.csv, но заполнен в b.csv. Я хочу сопоставить значения j k L столбцы, использующие " i " в качестве ключевого столбца из b.csv для a.csv файл
мое предложение было бы чем-то как этот (без использования defaultdict):
a_file = "path/to/a.csv"
b_file = "path/to/b.csv"
# read from file a.csv
with open(a_file) as f:
# skip headers
f.next()
# get first colum as keys
keys = (line.split(',')[0] for line in f)
# create empty dictionary:
d = {}
# read from file b.csv
with open(b_file) as f:
# gather headers except first key header
headers = f.next().split(',')[1:]
# iterate lines
for line in f:
# gather the colums
cols = line.strip().split(',')
# check to make sure this key should be mapped.
if cols[0] not in keys:
continue
# add key to dict
d[cols[0]] = dict(
# inner keys are the header names, values are columns
(headers[idx], v) for idx, v in enumerate(cols[1:]))
обратите внимание, что для разбора csv-файлов есть csv модуль.
обновление: для произвольной длины вложенного словаря перейдите в ответ.
используйте функцию defaultdict из коллекций.
высокая производительность: "если ключ не в dict" очень дорого, когда набор данных большой.
низкое обслуживание: сделайте код более читаемым и смогите легко быть расширено.
from collections import defaultdict
target_dict = defaultdict(dict)
target_dict[key1][key2] = val
для произвольных уровней nestedness:
In [2]: def nested_dict():
...: return collections.defaultdict(nested_dict)
...:
In [3]: a = nested_dict()
In [4]: a
Out[4]: defaultdict(<function __main__.nested_dict>, {})
In [5]: a['a']['b']['c'] = 1
In [6]: a
Out[6]:
defaultdict(<function __main__.nested_dict>,
{'a': defaultdict(<function __main__.nested_dict>,
{'b': defaultdict(<function __main__.nested_dict>,
{'c': 1})})})
важно помнить, что при использовании defaultdict и подобных вложенных модулей dict, таких как nested_dict, поиск несуществующего ключа может непреднамеренно создать новую запись ключа в dict и вызвать много хаоса. Вот пример Python3 с nested_dict.
import nested_dict as nd
nest = nd.nested_dict()
nest['outer1']['inner1'] = 'v11'
nest['outer1']['inner2'] = 'v12'
print('original nested dict: \n', nest)
try:
nest['outer1']['wrong_key1']
except KeyError as e:
print('exception missing key', e)
print('nested dict after lookup with missing key. no exception raised:\n', nest)
# instead convert back to normal dict
nest_d = nest.to_dict(nest)
try:
print('converted to normal dict. Trying to lookup Wrong_key2')
nest_d['outer1']['wrong_key2']
except KeyError as e:
print('exception missing key', e)
else:
print(' no exception raised:\n')
# or use dict.keys to check if key in nested dict.
print('checking with dict.keys')
print(list(nest['outer1'].keys()))
if 'wrong_key3' in list(nest.keys()):
print('found wrong_key3')
else:
print(' did not find wrong_key3')
вывод:
original nested dict: {"outer1": {"inner2": "v12", "inner1": "v11"}}
nested dict after lookup with missing key. no exception raised:
{"outer1": {"wrong_key1": {}, "inner2": "v12", "inner1": "v11"}}
converted to normal dict.
Trying to lookup Wrong_key2
exception missing key 'wrong_key2'
checking with dict.keys
['wrong_key1', 'inner2', 'inner1']
did not find wrong_key3