python怎么将四行代码换成一行啊