Как улучшить код?

Question

[Alan Kertz]

Вчера начал изучать Python. Оцените пожалуйста код:

import math
from operator import *
 
#Functions for the calculating
def calc():
    select = input("Select an operator: + - * / ** √\n")
    operators = ["+", "-", "*", "/", "**"]
    try:
        if select == "√":
            root = float(input("Enter a number..."))
            print("√", root,"=", math.sqrt(root))
        elif select in operators:
            num1 = float(input("Enter the first number...\n"))
            num2 = float(input("Enter the second number...\n"))
            if select == "+":
                print(num1, "+",num2,"=", add(num1, num2))
            elif select == "-":
                print(num1, "-",num2,"=", sub(num1, num2))
            elif select == "*":
                print(num1, "*",num2,"=",  mul(num1, num2))
            elif select == "/":
                print(num1, "/",num2,"=", div(num1, num2))
            elif select == "**":
                print(num1, "**",num2,"=", pow(num1, num2))
            else:
                print("Error, invalid input")
    except:
        print("Error, you need to enter a number")
 
#The main function which use a while loop
def main():
    i = 0
    while True:
        if i == 0:
            calc()
        #print('\nCalculation finished. Do another? (Y/N)')
        selector = input("Calculation finished. Do another? (Y/N)\n")
        if selector.lower() == 'n':
            break
        elif selector.lower() == 'y':
            calc()
        i = 1
 
#About the script, here I use the main function
print("This is a simple calculator on python")
main()

Буду благодарен за любую критику!

Answer 1

[kolomiec_artiom]

Ну, для того, чтобы визуально уменьшить код - я бы использовал вместо конструкции "if - else" словарь

Comments

[Alan Kertz]

Благодарю вас! Как считаете, для новичка, который день изучает Python это нормальный уровень?

[kolomiec_artiom]

Alan Kertz, да, вполне!)

Answer 2

[Владимир Куц]

Как улучшить код?

К примеру так:

import math
import operator

def calc():
    op = input("Select an operator: + - * / ** √\n")
    
    if op in ['√']:
        num = float(input('Enter a number: '))
        res = f'√{num} = {math.sqrt(num)}'
    else:
        num1 = float(input("Enter the first number...\n"))
        num2 = float(input("Enter the second number...\n"))
        res = {
           '+': f'{num1} + {num2} = {operator.add(num1, num2)}',
           '-': f'{num1} - {num2} = {operator.sub(num1, num2)}',
           '*': f'{num1} * {num2} = {operator.mul(num1, num2)}',
           '/': f'{num1} / {num2} = {operator.truediv(num1, num2)}',
           '**': f'{num1} ** {num2} = {operator.pow(num1, num2)}',
        }.get(op, 'Unknown operation')
    print(res)

if __name__ == '__main__':
    print("This is a simple calculator on python")
    while True:
        calc()
        if input("Calculation finished. Do another? (Y/N)\n").lower() != 'y':
            break

А можно и мой вариант улучшить.

Comments

[Alan Kertz]

Благодарю вас, извиняюсь, если я потратил ваше премя!

[Alan Kertz]

Как можно улучшить ваш вариант?

[bbkmzzzz]

Alan Kertz,

op = input("Select an operator: + - * / ** √\n")
if op in ['√']

Проверяем вхождение строки в список с одним элементом, который строка из одного символа. Зачем?
if op == '√':
Можно так

[Владимир Куц]

bbkmzzzz,

Зачем?

Задел на будущее... Если мне нужно добавить еще унарных операций - я просто сделаю так:

if op in ['√', '!']:
        num = float(input('Enter a number: '))
        res = {
             '√': f'√{num} = {math.sqrt(num)}',
             '!': f'!{num} = {math.factorial(num)}'
        }.get(op)
else:
    ...

При этом код не потеряет стройности и лаконичности. Это намек к вопросу о дальнейшем улучшении кода ))