Hackerrank python pandas