Python Tricks 若干

long

赵斌 — APRIL 29, 2015
在 python 代码中可以看到一些常见的 trick，在这里做一个简单的小结。

json 字符串格式化
在开发 web 应用的时候经常会用到 json 字符串，但是一段比较长的 json 字符串是可读性较差的，不容易看出来里面结构的。 这时候就可以用 python 来把 json 字符串漂亮的打印出来。

• 
  root@Exp-1:/tmp# cat json.txt
  
  
• 
  {"menu": {"breakfast": {"English Muffin": {"price": 7.5}, "Bread Basket": {"price": 20, "desc": "Assortment of fresh baked fruit breads and muffins"}, "Fruit Breads": {"price": 8}}, "drink": {"Hot Tea": {"price": 5}, "Juice": {"price": 10, "type": ["apple", "watermelon", "orange"]}}}}
  
  
• 
  root@Exp-1:/tmp#
  
  
• 
  root@Exp-1:/tmp# cat json.txt | python -m json.tool
  
  
• 
  {
  
  
• 
  "menu": {
  
  
• 
  "breakfast": {
  
  
• 
  "Bread Basket": {
  
  
• 
  "desc": "Assortment of fresh baked fruit breads and muffins",
  
  
• 
  "price": 20
  
  
• 
  },
  
  
• 
  "English Muffin": {
  
  
• 
  "price": 7.5
  
  
• 
  },
  
  
• 
  "Fruit Breads": {
  
  
• 
  "price": 8
  
  
• 
  }
  
  
• 
  },
  
  
• 
  "drink": {
  
  
• 
  "Hot Tea": {
  
  
• 
  "price": 5
  
  
• 
  },
  
  
• 
  "Juice": {
  
  
• 
  "price": 10,
  
  
• 
  "type": [
  
  
• 
  "apple",
  
  
• 
  "watermelon",
  
  
• 
  "orange"
  
  
• 
  ]
  
  
• 
  }
  
  
• 
  }
  
  
• 
  }
  
  
• 
  }
  
  
• 
  root@Exp-1:/tmp#
  
  

else 的妙用
在某些场景下我们需要判断我们是否是从一个 for 循环中 break 跳出来的，并且只针对 break 跳出的情况做相应的处理。这时候我们通常的做法是使用一个 flag 变量来标识是否是从 for 循环中跳出的。 如下面的这个例子，查看在 60 到 80 之间是否存在 17 的倍数。

• 
  flag = False
  
  
• 
  for item in xrange(60, 80):
  
  
• 
  if item % 17 == 0:
  
  
• 
  flag = True
  
  
• 
  break
  
  
• 
  
  
• 
  if flag:
  
  
• 
  print "Exists at least one number can be divided by 17"
  
  

其实这时候可以使用 else 在不引入新变量的情况下达到同样的效果

• 
  for item in xrange(60, 80):
  
  
• 
  if item % 17 == 0:
  
  
• 
  flag = True
  
  
• 
  break
  
  
• 
  else:
  
  
• 
  print "exist"
  
  

setdefault 方法
dictionary 是 python一个很强大的内置数据结构，但是使用起来还是有不方便的地方，比如在多层嵌套的时候我们通常会这么写

• 
  dyna_routes = {}
  
  
• 
  method = 'GET'
  
  
• 
  whole_rule = None
  
  
• 
  # 一些其他的逻辑处理
  
  
• 
  ...
  
  
• 
  if method in dyna_routes:
  
  
• 
  dyna_routes[method].append(whole_rule)
  
  
• 
  else:
  
  
• 
  dyna_routes[method] = [whole_rule]
  
  

其实还有一种更简单的写法可以达到同样的效果

　self.dyna_routes.setdefault(method, []).append(whole_rule)

或者可以使用 ｀collections.defaultdict｀ 模块

• 
  import collections
  
  
• 
  dyna_routes = collections.defaultdict(list)
  
  
• 
  ...
  
  
• 
  dyna_routes[method].append(whole_rule)
  
  

本文作者系OneAPM工程师赵斌 ，想阅读更多好的技术文章，请访问OneAPM官方技术博客。