pta问题电价分阶问题

某电价规定:月用电量在150千瓦时及以下部分按每千瓦时0.4463元收费,月用电量在151~400千瓦时的部分按每千瓦时0.4663元收费,月用电量在401千瓦时及以上部分按每千瓦时0.5663元收费。
请编写一个程序,根据输入的月用电量(单位以千瓦时计),按该电价规定计算出应缴的电费(单位以元计)。
输入格式:
首先输入一个正整数T,表示测试数据的组数,然后是T组测试数据。对于每组测试,输入一个整数n(0≤n≤10000),表示月用电量。
输出格式:
对于每组测试,输出一行,包含一个实数,表示应缴的电费。结果保留2位小数。
输入样例:
1
267
输出样例:
121.50

import pandas as pd


def main():
    DataList = []
    
    G_number = int(input('输入测试组:'))
    
    for i in range(G_number):
        Electricity = int(input('输入用电量:'))
        DataList.append(Electricity)
    
    for i in range(0,len(DataList)):
        if 0 < DataList[i] <= 150:
            E_values = DataList[i] * 0.4463
        elif 150 < DataList[i]  <= 400:
            E_values =  (150 * 0.4463) + ((DataList[i] - 150) * 0.4663)
        elif DataList[i] > 400:
            E_values =  (150 * 0.4463) + ((400-150) * 0.4663) + ((DataList[i] - 400) * 0.5663)
        print('第{}组数据 , 用电量:{} , 应缴的电费为:{:.2f}'.format(i+1,DataList[i],E_values)) 
    

if '__main__' == __name__:
    main()