Я работаю с python3, где у меня есть список словарей

1

У меня есть список словарей, которые содержат имя_пользователя, имя счета и город

question = [{
        "count": 2,
        "city": "Pune",
        "job_NAME": "clerk"
    }, {
        "count": 1,
        "city": "kolkata",
        "job_NAME": "manager"
    }, {
        "count": 2,
        "city": "Mumbai",
        "job_NAME": "accountant"
    }, {
        "count": 1,
        "city": "New Delhi",
        "job_NAME": "clerk"
    }]

что я хочу, если job_name одинаково в этом случае "клерк", граф и город должны быть в одном списке под ключевыми "заданиями",

   answer = [{
    "job_name": "clerk",
    "jobs": [{
        "city": "pune",
        "count": 2
    }, {
        "city": "New Delhi",
        "count": 1
    }]
}, {
    "job_name": "manager",
    "jobs": [{
        "city": "kolkata",
        "count": 1
    }]
}, {
    "job_name": "accountant",
    "jobs": [{
        "city": "Mumbai",
        "count": 2
    }]
}]

если в списке вопросов есть дубликат вакансий, в этом случае клерк его города и счет должны быть добавлены в список "jobs".

я был бы очень благодарен за любую предоставленную помощь!

заранее спасибо

Теги:
python-3.x
pymongo

3 ответа

1

Я использую defaultdict(list) чтобы сначала сгруппировать записи из списка ввода, а затем во второй итерации я установил результат:

from collections import defaultdict

groups = defaultdict(list)

for jobs in question:
    job_name = jobs["job_NAME"]
    del jobs["job_NAME"]
    groups[job_name].append(jobs)

answer = [
    dict(job_name=job_name, jobs=jobs) for job_name, jobs in groups.items()
]

# alternative with classic loop
answer = []
for job_name, jobs in groups.items():
    answer.append(dict(job_name=job_name, jobs=jobs))
1

Использование collections.defaultdict

Пример:

from collections import defaultdict

d = defaultdict(list)
question = [{'count': 2, 'city': 'Pune', 'job_NAME': 'clerk'}, {'count': 1, 'city': 'kolkata', 'job_NAME': 'manager'}, {'count': 2, 'city': 'Mumbai', 'job_NAME': 'accountant'}, {'count': 1, 'city': 'New Delhi', 'job_NAME': 'clerk'}]

for i in question:
    d[i["job_NAME"]].append({"city": i["city"], "count": i["count"]})

answer = [{"job_name": k, "jobs": v} for k, v in d.items()]
print(answer)

Выход:

[{'job_name': 'clerk',
  'jobs': [{'city': 'Pune', 'count': 2}, {'city': 'New Delhi', 'count': 1}]},
 {'job_name': 'accountant', 'jobs': [{'city': 'Mumbai', 'count': 2}]},
 {'job_name': 'manager', 'jobs': [{'city': 'kolkata', 'count': 1}]}]
0

Один лайнер:

from itertools import groupby
from operator import itemgetter

answer = [{'job_name': k, 'jobs': [{f: v for f, v in d.items() if f != 'job_NAME'} for d in g]} 
          for k, g in groupby(sorted(question, key=itemgetter('job_NAME')), key=itemgetter('job_NAME'))]

Результат:

[{'job_name': 'accountant', 'jobs': [{'count': 2, 'city': 'Mumbai'}]},
 {'job_name': 'clerk',
  'jobs': [{'count': 2, 'city': 'Pune'}, {'count': 1, 'city': 'New Delhi'}]},
 {'job_name': 'manager', 'jobs': [{'count': 1, 'city': 'kolkata'}]}]

Сортируйте и группируйте question по job_NAME а затем просто job_NAME ответ в виде списка job_name с job_name и jobs.

Ещё вопросы

Сообщество Overcoder
Наверх
Меню