支持拼音首字母检索
#!/usr/bin/env python3
# encoding: utf-8
import pinyin
import sys
import re
query = sys.argv[1]
pattern = re.compile(query)
for line in sys.stdin:
p = pinyin.get_initial(line).replace(' ', '')
if re.search(pattern, p):
print(line)