HackerRank Compress the String!

###题意：给你一个由数字组成的串，输出每段连续出现的字符，及其出现次数，最少出现次数可以是1次

###思路：记录第一次出现下标，每次将当前的和之前一个字符比较看是否一样，一样+1，不一样初始化为1.（这里还是沿用c++的思路，所以代码比较长），见代码：

if __name__ in '__main__':
    str = input()
    #由于有重复的键不能申请dict，目前能想到的只有list
    #lst为每个键出现的次数
    #key为键
    lst = []
    key = []
    lst.append(1)
    key.append(str[0])
    idx = 0

    #每次判断当前字符和前一个是否一样，并且记录第一次出现这个字符的位置（下标）
    for i in range(1, len(str)):
        if str[i] == str[i-1]:
            lst[idx] += 1
        else:
            lst.append(1)
            key.append(str[i])
            idx += 1
            
    #最后一个空格会在样例结束多输出一个，这个卡的不严没影响
    for i in range(0, len(lst)):
        print((lst[i], int(key[i])), end = " ")

下面代码使用python当中的groupby函数，在我目前理解他就是一个自定义分组函数，这样代码非常短，他会自动分组，按照连续相同的键分组，并可以计算连续的长度，见代码：

from itertools import groupby
for a, b in groupby(input()):
    print((len(list(b)), int(a)), end = " ")