PTA 乙级 1051 复数乘法 (15分) Python

突然想用python了

用C++的话，注意计算出a,b后像 -0.005 < a < 0.005, -0.005 < b < 0.005 这种情况会保留两位小数时，会直接输出0.00，应该输出0（测试点2,3），所以需要判定a,b的范围

python直接利用round提前进行小数点的保留即可

import math
num = input().split(' ')
r1 = eval(num[0])
p1 = eval(num[1])
r2 = eval(num[2])
p2 = eval(num[3])
a = round(r1 * r2 * math.cos(p1 + p2), 2)   #a,b提前进行小数保留
b = round(r1 * r2 * math.sin(p1 + p2), 2)
if b == 0:          #当a,b为0时，必须输出0，不可以输出0.0(测试点2,3)
    b = 0
elif a == 0:
    a = 0
if b >= 0:
    print("{0:.2f}+{1:.2f}i".format(a, b))
else:
    print("{0:.2f}-{1:.2f}i".format(a, abs(b)))