Python Simple Calculator with Square Root

This guide will walk you through creating a simple Python calculator that includes square root functionality. We'll cover the basic structure, how to implement mathematical operations, and how to add the square root feature. By the end, you'll have a functional calculator that you can expand with additional features.

Introduction

Python is a versatile programming language that's great for creating small utility programs like calculators. A simple calculator can help you understand basic Python concepts such as user input, conditional statements, and functions. Adding square root functionality demonstrates how to work with mathematical operations and error handling.

Prerequisites

To follow along with this guide, you should have:

Basic knowledge of Python programming
Python installed on your computer
A text editor or IDE for writing Python code

Creating a Basic Calculator

The foundation of our calculator will be a simple program that can perform basic arithmetic operations. We'll start by creating a function that takes two numbers and an operator as input, then returns the result of the operation.

Basic Calculator Structure

def basic_calculator(num1, num2, operator):
    if operator == '+':
        return num1 + num2
    elif operator == '-':
        return num1 - num2
    elif operator == '*':
        return num1 * num2
    elif operator == '/':
        if num2 == 0:
            return "Error: Division by zero"
        return num1 / num2
    else:
        return "Error: Invalid operator"

This function checks the operator and performs the corresponding arithmetic operation. It includes basic error handling for division by zero and invalid operators.

Adding Square Root Functionality

To add square root functionality, we'll extend our basic calculator. Python's math module provides a square root function that we can use. We'll need to import this module and add a new condition to our calculator function.

Square Root Implementation

import math

def calculator(num1, num2=None, operator=None):
    if operator == 'sqrt':
        if num1 < 0:
            return "Error: Cannot calculate square root of negative number"
        return math.sqrt(num1)
    elif num2 is not None and operator is not None:
        return basic_calculator(num1, num2, operator)
    else:
        return "Error: Invalid input"

This enhanced function first checks if the operation is a square root. If so, it calculates the square root using math.sqrt(). For other operations, it falls back to the basic calculator function. It includes error handling for negative numbers in square root calculations.

Complete Code Example

Here's the complete code for our Python calculator with square root functionality:

Complete Calculator Code

import math

def basic_calculator(num1, num2, operator):
    if operator == '+':
        return num1 + num2
    elif operator == '-':
        return num1 - num2
    elif operator == '*':
        return num1 * num2
    elif operator == '/':
        if num2 == 0:
            return "Error: Division by zero"
        return num1 / num2
    else:
        return "Error: Invalid operator"

def calculator(num1, num2=None, operator=None):
    if operator == 'sqrt':
        if num1 < 0:
            return "Error: Cannot calculate square root of negative number"
        return math.sqrt(num1)
    elif num2 is not None and operator is not None:
        return basic_calculator(num1, num2, operator)
    else:
        return "Error: Invalid input"

# Example usage
print(calculator(9, operator='sqrt'))  # Output: 3.0
print(calculator(4, 2, '+'))            # Output: 6
print(calculator(10, 2, '/'))          # Output: 5.0

This complete code includes both the basic calculator functionality and the square root feature. You can run this code in any Python environment to see it in action.

Usage Examples

Let's look at some practical examples of how to use our calculator:

Operation	Code	Result
Square root of 16	`calculator(16, operator='sqrt')`	4.0
Addition: 5 + 3	`calculator(5, 3, '+')`	8
Division: 10 / 2	`calculator(10, 2, '/')`	5.0
Square root of -1 (error case)	`calculator(-1, operator='sqrt')`	"Error: Cannot calculate square root of negative number"

These examples demonstrate both successful operations and error cases. The calculator handles both positive and negative numbers appropriately for square root calculations.

Frequently Asked Questions

Can I use this calculator for complex numbers?: No, this calculator only handles real numbers. For complex numbers, you would need to use Python's cmath module instead of math.
What happens if I enter an invalid operator?: The calculator will return an "Error: Invalid operator" message. It only accepts '+', '-', '*', '/', and 'sqrt' as valid operators.
How do I handle division by zero?: The calculator checks for division by zero and returns an appropriate error message. It's important to handle this case to prevent program crashes.
Can I modify this calculator to include more functions?: Yes, you can easily extend this calculator by adding more conditions to the calculator function. For example, you could add exponentiation or logarithms.
Is there a way to make this calculator interactive?: Yes, you could create a simple command-line interface using input() functions to make the calculator interactive. For a more advanced interface, you could use a GUI library like Tkinter.