Что нужно сделать
Спустя некоторое время заказчик попросил вас немного изменить скрипт для криптографии: теперь индексы элементов должны быть простыми числами.
Напишите функцию, которая возвращает список элементов итерируемого объекта (кортежа, строки, списка, словаря), у которых индекс — это простое число. Для проверки на простое число напишите отдельную функцию is_prime
. Основной код оставьте пустым (используйте его только для тестирования).
Дополнительно: сделайте так, чтобы основная функция состояла только из оператора return и так же возвращала список.
Пример вызова функции:
Пример вызова функции:
print(crypto([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]))
Ответ в консоли: [2, 3, 5, 7]
Пример вызова функции:
print(crypto('О Дивный Новый мир!'))
Ответ в консоли: ['Д', 'и', 'н', 'й', 'в', 'й', 'р']
Что оценивается
- Результат вычислений корректен.
- Весь функционал программы описан в функциях.
- Переменные и функции имеют значащие имена, не только a, b, c, d
def is_prime(number):
if number > 1:
for j in range(2, int(number / 2) + 1):
if (number % j) == 0:
return False
else:
return True
else:
return False
def cryptoprime(elem):
answer = []
for i, v in enumerate(elem):
if is_prime(i) == True:
answer.append(v)
return answer