各位相加（Python）

258. 各位相加

给定一个非负整数 num，反复将各个位上的数字相加，直到结果为一位数。

示例:

输入: 38

输出: 2

解释: 各位相加的过程为：3 + 8 = 11, 1 + 1 = 2。 由于 2 是一位数，所以返回 2。

def func(num):
    # while len(str(num)) != 1:
    #     j = 0
    #     for i in str(num):
    #         j += int(i)
    #     num = j
    # return num
    if num == 0:
        return 0
    if num % 9 == 0:
        return 9
    else:
        return num % 9

print(func(38))