python - 这段函数如何改写比较优雅?

问题描述

如图:

其实就是多了一个判断而已 ( 但是我因此需要复制2段一样的代码 )

具体代码如下:

def get_all_ranks(df,xm): if xm == '所有':for i in df.index: #如果项目登录所有(就不做判断了,直接全部都查询!!!!) url = df['域名'][i] kw = df['关键词'][i] #查询一下收录 df['收录'][i] = Baidu().baidu_sites(url) #如果没有收录 if df[’收录’][i] != 0:#就不查排名了df['排名'][i]=df['手机排名'][i] = -1 #否则就更新一下排名 else:df['排名'][i] = Baidu().rank(kw,url)[0]df['手机排名'][i]=Baidu().rank(kw,url,zd='m')[0] print ('%s,关键词:%s,pc排名:%s,手机排名%s' % (url,kw,df['排名'][i],df['手机排名'][i]))return dfelse:for i in df.index: if df['项目'][i] == xm:url = df['域名'][i]kw = df['关键词'][i]#查询一下收录df['收录'][i] = Baidu().baidu_sites(url)#如果没有收录if df[’收录’][i] != 0: #就不查排名了 df['排名'][i]=df['手机排名'][i] = -1#否则就更新一下排名else: df['排名'][i] = Baidu().rank(kw,url)[0] df['手机排名'][i]=Baidu().rank(kw,url,zd='m')[0]print ('%s,关键词:%s,pc排名:%s,手机排名%s' % (url,kw,df['排名'][i],df['手机排名'][i]))return df

有高手会改写得优雅一点吗? 好像有个装饰器的东西但是不太会用~

问题解答

xm的值应该在循环中判断,并且xm为所有时并不是特殊到值得单独处理。

for i in df.index: if xm == ’所有’ or xm == i:#do something