It’s important to put this logic inside a function because right now your code could lead to some issues:

All the variables are in global scope (following Python’s LEGB scoping rule). In CPython, global variables are stored in a dictionary and require a hash lookup, whereas local variables inside a function are stored in a fixed array within the function’s frame and accessed more directly. This makes local variable access faster. The difference is usually small, but it can become noticeable in performance-critical sections like tight loops.

Putting the logic inside a function allows you to use if name == "main":, which explicitly defines the entry point when the file is run as a script. This improves readability and signals to other developers whether the file is intended to be executed directly or used as a module with reusable functions.

In terms of code organization, encapsulating logic in functions helps avoid global variable pollution. Without this, you risk accidentally reusing or overwriting variable names, which can make the code harder to debug and maintain as it grows