使用python将罗马字转换为对应的阿拉伯数字

#encoding:utf-8
#将罗马数字转换为阿拉伯数字
dict={'I':1,'V':5,'X':10,'L':50,'C':100,'D':500,'M':1000} #罗马单个字母的数字值
dict2={'IV':4,'IX':9,'XL':40,'XC':90,'CD':400,'CM':900} #几个特殊情况
dict3=('IV','IX','XL','XC','CD','CM')
#现在要求输入罗马字母，输出对应的数字值
n=input()
p=0
if n in dict3:
    p=dict2[n]
else:
    for i in n:
        for k,v in dict.items():
            if i  == k:
                p+=v
print(p)