simplify by using list comprehensions
Merge pull request #5067 from hulu/ps-list-comp

ps module: simplify by using list comprehensions