Change Python logger format in AWS Lambda

change_lambda_logger_format.py

# Python logger in AWS Lambda has a preset format. To change the format of the logging statement, remove the logging handler 
# and add a new handler with the required format

logger = logging.getLogger()
for h in logger.handlers:
  logger.removeHandler(h)

h = logging.StreamHandler(sys.stdout)

# use whatever format you want here
FORMAT = '%(asctime)s %(message)s'
h.setFormatter(logging.Formatter(FORMAT))
logger.addHandler(h)
logger.setLevel(logging.INFO)