帮我用python写一个阶梯电费计算

好的，以下是一个简单的阶梯电费计算的 Python 代码：

def electricity_bill(units):
    if units <= 50:
        bill = units * 0.5
    elif units <= 150:
        bill = 25 + (units - 50) * 0.75
    elif units <= 250:
        bill = 100 + (units - 150) * 1.2
    else:
        bill = 220 + (units - 250) * 1.5
    return bill

# 测试代码
print(electricity_bill(30))   # 应输出：15.0
print(electricity_bill(100))  # 应输出：62.5
print(electricity_bill(200))  # 应输出：190.0
print(electricity_bill(300))  # 应输出：335.0

该函数 `electricity_bill` 接受一个参数 `units`，表示用电量，根据不同的用电量，计算出对应的电费。其中：

- 如果用电量不超过 50 度，电费为用电量乘以 0.5 元；
- 如果用电量在 50~150 度之间，前 50 度电费为 25 元，超过 50 度的电费为用电量减去 50 度后的部分乘以 0.75 元；
- 如果用电量在 150~250 度之间，前 150 度电费为 100 元，超过 150 度的电费为用电量减去 150 度后的部分乘以 1.2 元；
- 如果用电量超过 250 度，前 250 度电费为 220 元，超过 250 度的电费为用电量减去 250 度后的部分乘以 1.5 元。

你可以根据需要修改阶梯电价的比例和阶梯划分的值。