복잡하고 어지러운 코드의 최적화 프롬프트

Python에서 코드 옵티마이저 역할을 해줘. 코드를 더 효율적으로 만들어줄 수 있어?

def calculate_sum(n):
result = 0
for i in range(n):
if i % 3 == 0 or i % 5 == 0:
result += i
return result

def main():
num = input("Enter a number: ")
num = int(num)
sum = calculate_sum(num)
print("The sum of all numbers less than", num, "that are divisible by 3 or 5 is:", sum)

if name == 'main':
main()

주어진 코드는 이미 충분히 간단하고 직관적이지만, 좀 더 최적화할 수 있는 방법이 있습니다. 다음은 더욱 간단한 코드 예시입니다:

def calculate_sum(n):
return sum(i for i in range(n) if i % 3 == 0 or i % 5 == 0)

if name == 'main':
num = int(input("Enter a number: "))
print("The sum of all numbers less than", num, "that are divisible by 3 or 5 is:", calculate_sum(num))

이 코드는 `calculate_sum` 함수를 한 줄로 간략화하고, `main` 함수를 제거하였습니다

`calculate_sum` 함수는 `range(n)`에서 생성되는 모든 자연수 `i`에 대해서 `i`가 3 또는 5의 배수인 경우 `i`를 더한 값을 반환합니다. 이를 위해 `sum` 함수와 generator expression을 사용합니다.

마지막으로, `if __name__ == '__main__':` 문은 이 파일이 다른 모듈에서 import 될 때는 실행되지 않고, 직접 실행될 때만 `calculate_sum` 함수를 실행하도록 합니다.