Возможно в Python сделать многоуровневые if / как сделать задачку с калькулятором Stepik?

Question

[MelomanIk]

b = float(input())
x = input()
if x == "+" :
 print(a+b)
elif x =="-" :
 print(a-b)
elif x =="*" :
 print(a*b)
elif x =="pow" :
 print(a**b)
elif x=='/':
    if b=='0':
     print('Деление на 0!')
    else:
     print(a/b)
elif x=='mod':
    if b=='0':
     print('Деление на 0!')
    else:
      print(a//b)
elif x=='div':
    if b=='0':
     print('Деление на 0!')
    else:
     print(a%b)
# не робит це

a = float(input())

Comments

[MelomanIk]

Hellayas, минуту

[MelomanIk]

Hellayas, все в норме ?

[soremix]

сама задача?

[MelomanIk]

SoreMix, калькулятор
Главное, чтобы на if и при делении на ноль print('Деление на 0!')

Answer 1

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

def calc(a,b,op):
    if op in ['/', 'mod', 'div'] and b == 0:
        return 'Деление на 0!'
    else:
        return({
              '+': a+b,
              '-': a-b,
              '*': a*b,
              '/': a/b,
              'mod': a//b,
              'div': a%b,
            }.get(op, 'Неизвестная операция!'))

print(calc(1,2,'+'))
print(calc(1,2,'/'))
print(calc(2,0,'mod'))
print(calc(0,2,'$'))

3
0.5
Деление на 0!
Неизвестная операция!

Comments

[MelomanIk]

Вроде робит, но только простите, я не могу понять про input()
Вы не могли бы обьяснить
Это всё нужно на платформе, которая САМА вводит числа в input

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

MelomanIk,
ну добавьте ввод переменных:

a = float(input())
b = float(input())
x = input()

calc(a, b, x)

[MelomanIk]

Владимир Куц, пока не робит
Видимо Stepik работает только с if,
Так ваш код каждый введённый + воспримет как функцию ?

Answer 2

[Лариса .•º]

да, но я бы сделала так,

import operator

action = {
    "+": operator.add,
    "-": operator.sub,
    "/": operator.truediv,
    "*": operator.mul,
    "**": operator.pow
    }
# .....

b = float(input())
a = float(input())
x = input()
act= action.get(x, None)
if act:
   try:
     answer = act(a, b)
   except  ZeroDivisionError as e:
       print(u"Деление на 0!")
   else:
     print(answer)

Comments

[MelomanIk]

Спасибо, но Главное, чтобы на if и при делении на ноль print('Деление на 0!')

Answer 3

[Антон Федорян]

def addition(a, b):
    return a + b

def subtraction(a, b):
    return a - b

operations = {
    '+': addition,
    '-': subtraction
}

print(operations['+'](5, 7));
print(operations['-'](9, 4));