Как совместить 2 цикла и будет ли это оптимизировать программу?

Question

[Олег Париев]

Есть такой кусок кода с циклами:

inter_mass_n = {item : 0 for i in mass_q for item in range(i[0]-1, i[1])}

for i in mass_q:
    for j in range(i[0]-1, i[1]):
        inter_mass_n[j] += 1

Де-факто, программа выполняет лишнюю работу, потому хотелось бы сразу делать +=1 при создании словаря. Можно ли такое организовать и будет ли такой вариант работать быстрее?

Answer 1

[dmshar]

Де-факто программа выполняет непонятно что. Например

inter_mass_n = {item : 0 for i in mass_q for item in range(i[0]-1, i[1])}

создает словарь со всеми нулевыми значениями.
А тут просто увеличивает эти нули на единицу.

for i in mass_q:
    for j in range(i[0]-1, i[1]):
        inter_mass_n[j] += 1

Что-бы это значило? Почему сразу не присваивать единицы?

Comments

[Олег Париев]

Вообще весь код нужен для решения олимпиадной задачи. Но если выложить всё решение, то будут стараться решить задачу за меня. А мне интересно можно ли как-то оптимизировать эту часть кода. Чтобы не 2 раза проходиться по массиву, а только один

[Олег Париев]

mass_q - массив с отрезками, на которых надо складывать числа. Эта часть кода считает сколько раз встречается то или иное число в отрезках. И потом уже идёт подсчёт ответа

[dmshar]

Так это три (или две) разные части кода? Инициализация словаря, и отдельно что-то с чем-то складывается? Тогда что с чем вы объединять хотите?

[Олег Париев]

dmshar,

inter_mass_n = {item : 0 for i in mass_q for item in range(i[0]-1, i[1])}

for i in mass_q:
    for j in range(i[0]-1, i[1]):
        inter_mass_n[j] += 1

Это одна часть кода, которую я хочу сократить, т.к. 2 раза проходить по словарю - слишком много времени занимает... Потому задал вопрос как 2 цикла сократить до одного

[o5a]

Олег Париев, Я так понимаю вы приводите какой-то выдуманный аналог реального кода и хотите по нему подсказку для своего реального кода. Это какой-то испорченный телефон получается.

Если речь просто про то, как обращаться к словарю без инициализации, то используйте defaultdict

from collections import defaultdict

inter_mass_n = defaultdict(int)

for i in mass_q:
    for j in range(i[0]-1, i[1]):
        inter_mass_n[j] += 1

или dict.get()

inter_mass_n = dict()

for i in mass_q:
    for j in range(i[0]-1, i[1]):
        inter_mass_n[j] = inter_mass_n.get(j,0) + 1

[Олег Париев]

o5a, спасибо большое. Ваш ответ помог

[dmshar]

Олег Париев, Я же вам сразу написал: Почему сразу не присваивать единицы?
Вы начали рассказывать про то, что сначала делаете, что потом, и ожидали как оказалось, что-бы кто-то это написал за вас. Вместо того, чтобы вчитаться в ответ, самостоятельно подумать, самостоятельно исправить код и получить кайф от того, что вы это сделали самостоятельно. А за одно и чему-то научились.
Но дело ваше. Продолжайте жить на подсказках.

[Олег Париев]

dmshar, видимо, мы вас друг с другом не так поняли -_-

[o5a]

dmshar,

Я же вам сразу написал: Почему сразу не присваивать единицы?

Насколько понял, автору не нужно было именно "присваивать единицы". Ему нужны были какие-то свои манипуляции по словарю, просто он не знал, как его "инициализировать", а без этого не мог обратиться к несозданному ключу.