Как умножить списки разной длины в python?

Question

[Олег]

Надо создать список(из N элементов) и найти произведение элементов, индексы которых находятся в файле(100 штук).
С созданием списка и расшифровкой файла я справился. А вот при умножении упираюсь в ошибку "IndexError: list index out of range", ее победить не получилось.
То что написал прилагаю

N = int(input('Enter N: '))

import random

rnd_list = []
for i in range(0, N):
    rnd_list.append(random.randint(-N, N))
print(rnd_list)

with open("indexes.txt") as data: index_list = [int(line.strip()) for line in data]
print(index_list)

mult = 1
for i in index_list:
    mult *= rnd_list[index_list[i]]
print(mult)

Comments

[Модератор]

Фрагменты кода надо размещать в виде текста и оборачивать тэгом code для корректного отображения. Удобно делать кнопкой </>
Это обязательно, см.п.3.8 Регламента.
Сюда же относится traceback, ввод и вывод в консоли и другая структурированная текстовая инфа.
Также обратите внимание на п.3.4

Answer 1

[Алан Гибизов]

mult *= rnd_list[i]
или так:

for indx, value in enumerate(rnd_list):
    if indx in index_list:
        mult *= value

Comments

[Олег]

mult *= rnd_list[i]
Этот вариант упирается в ошибку list index out of range

второй вариант в ошибку не упирается, но считает не верно.
к примеру рандомно сгенерированный лист у меня получился:
[-3, -2, 4, 4, 4]
при введении N = 5,
в файле с индексами у меня есть минимальные индексы, до 5 которые:
0, 3, -4,
Получается программа должна перемножить элементы с этими индексами, но на выходе она мне выдала: -12, т.е. перемножив -3[0] на 4[3], а индекс -4 она пропустила.

UPD погонял программу увеличивая длинну списка, положительные индексы она просчитывает корректно, а вот все отрицательные пропускает.

[Алан Гибизов]

Олег, мне кажется, вы либо неверно описали задачу, либо неверно интерпретируете ее.
Я понял задачу так:
У вас есть некий файл с числами, которые надо интерпретировать как индексы. И надо перемножить те значения из списка случайных чисел, индексы которых есть в списке индексов.

При чем тут

в файле с индексами у меня есть минимальные индексы, до 5 которые

я не понимаю.

[Олег]

Простите если что-то не уточнил. Вы все верно понимаете. Файл с уже готовыми индексами, т.е. не я их генерирую, а уже составленный список в котором присутствуют и отрицательные и положительные индексы.
Программа проходит только по положительным игнорирую отрицательные.

[Алан Гибизов]

Олег,

import random
index_list = [1, 3, 5]
rnd_list = []
N = 7
for i in range(0, N):
    rnd_list.append(random.randint(-N, N))
print(rnd_list)
mult = 1
for indx, value in enumerate(rnd_list):
    if indx in index_list:
        mult *= value
print(mult)

Вот я вместо вашего файла с индексами взял тупо 1, 3, 5. N взял просто 7, можно брать любой.
Что не так? Подставьте свои значения N и индексов, при которых код работает неверно?

[Олег]

Алан Гибизов, Согласен, так работает, и по моему списку индексов работает, за исключением отрицательных индексов.
Если Вы в Вашем коде, а именно:
index_list = [1, 3, 5]
замените 1 на -1, то этот индекс будет игнорироваться.

[Алан Гибизов]

Олег, а как вы себе представляете отрицательный индекс? С конца? Ну так сделайте, как считаете нужным. Реализуйте отдельную обработку отрицательных индексов через if/elif или через dict или через match/case.

Answer 2

[Vindicar]

for i in index_list:
    mult *= rnd_list[i]

Должно быть правильно. Если вываливается IndexError - значит, в index_list есть индексы, которые не вписываются в rnd_list - либо >N-1, либо <-N. Уточняй условие задачи, что делать, если встретились такие индексы.
Ну и можно для уточнения ситуации сделать так:

N = len(rnd_list)
for i in index_list:
    try:
        value = rnd_list[i]
    except IndexError:
        print(f'Не удалось взять элемент с индексом {i}, он не в диапазоне {-N}...{N-1}. Игнорирую индекс.')
    else:
        mult *= value

Comments

[Олег]

for i in index_list:
    mult *= rnd_list[i]

Сам смысл кода верный, но за счет того что индексов 100, а я к примеру беру список из 10 элементов, он улетает в ошибку.

Уточняй условие задачи, что делать, если встретились такие индексы.

Задача чтобы при любой длине списка, при вводе N программа не улетала в ошибку и не выдавала лишние строки из Вашего второго примера.
Как я это понимаю, задали любую длину из любых чисел, программа пробежала по индексом и определила какие подходят под эту длину, а какие нет, перемножила элементы из rnd_list и выдала результат.
Должно быть игнорирование разницы в длине.
Именно это я победить и не могу =(

[Vindicar]

Олег, нет, ты вообще не понял в чём проблема.

за счет того что индексов 100, а я к примеру беру список из 10 элементов, он улетает в ошибку.

Вот тебе пример "на живую".

Python 3.10.4 (tags/v3.10.4:9d38120, Mar 23 2022, 23:13:41) [MSC v.1929 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> index_list = [1, 0, 2, 2, 1, 0, 0, 0, 1, 2]
>>> rnd_list = [2, 3, 4]
>>> for i in index_list:
...     print(rnd_list[i], end=' ')
...
3 2 4 4 3 2 2 2 3 4 
>>>

Как видишь, код прекрасно выбрал элементы из rnd_list по индексам в index_list, несмотря на то что index_list длиннее.
Неважно, сколько у тебя индексов. Важно, какие у них значения.
Так что ошибка у тебя возникает именно из-за того, что значения индексов в файле недопустимые, а не из-за того, что их много. Введи в файл один индекс 100500, и у тебя программа тоже не будет работать, если длина rnd_list меньше 100500.
Предусмотреть этот сценарий необходимо, а потому нужно определить, что программа должна делать в этом случае: останавливаться с ошибкой, выводить сообщение и игнорировать элемент, молча игнорировать элемент, или что-то ещё.