在Python中,你可以使用条件语句和列表(数组)来实现一个简单的计算器。以下是一个示例代码,展示了如何根据用户输入的操作符对两个数字进行加法、减法、乘法和除法操作:
```python
def simple_calculator(operation, num1, num2):
if operation == 'add':
return num1 + num2
elif operation == 'subtract':
return num1 - num2
elif operation == 'multiply':
return num1 * num2
elif operation == 'divide':
if num2 == 0:
return "Error: Division by zero is not allowed."
else:
return num1 / num2
else:
return "Error: Invalid operation."
示例使用
print(simple_calculator('add', 10, 20)) 输出:30
print(simple_calculator('subtract', 10, 20)) 输出:-10
print(simple_calculator('multiply', 10, 20)) 输出:200
print(simple_calculator('divide', 10, 20)) 输出:0.5
print(simple_calculator('divide', 10, 0)) 输出:"Error: Division by zero is not allowed."
print(simple_calculator('unknown', 10, 20)) 输出:"Error: Invalid operation."
这个函数接受三个参数:操作符(字符串),第一个数字(整数或浮点数)和第二个数字(整数或浮点数)。根据操作符的不同,函数执行相应的数学运算并返回结果。如果操作符无效或尝试除以零,函数将返回一个错误消息。
如果你需要处理更复杂的数组操作,比如对数组中的每个元素执行相同的操作,你可以使用列表推导式或循环。例如,以下代码将列表中的每个元素乘以2:
```python
numbers = [1, 2, 3, 4, 5]
doubled_numbers = [num * 2 for num in numbers]
print(doubled_numbers) 输出:[2, 4, 6, 8, 10]