Застрял после регистрации предупреждения, как заставить продолжение?

Question

У меня есть скрипт Python, который перебирает кадры данных для выполнения определенных манипуляций.Я использовал модуль регистрации для выдачи предупреждения, когда он обнаруживает пропущенные данные.По умолчанию он должен выдать предупреждение, а затем продолжить, но он застрял там.Есть ли способ заставить его продолжить после регистрации предупреждения?

if ctype == 'color': 
    if id not in data.columns:
        logger.warn("Data unavailable")
        return None, atuple
    data=data * 3

Я сталкиваюсь с экземпляром, где идентификатор не находится в data.columns, и код застрял там, не переходя к следующемустрока.Есть ли способ заставить его двигаться вперед?

Answer 1

Проблема заключается в операторе return, который завершит вызов функции и передаст результат туда, откуда он был вызван. Функция может только return значения один раз за вызов. Если вы хотите, чтобы функция продолжалась, вы можете либо собрать свои переменные и вернуть их в конце, либо использовать ключевое слово yield, которое будет возвращать значение при каждом попадании в строку.