帮我写一个水仙花数python代码

当一个三位数，它的每个位上的数字的立方和等于它本身，我们称这个数为水仙花数。例如：153 = 1^3 + 5^3 + 3^3，所以153就是一个水仙花数。

下面是一个Python代码实现：

for num in range(100, 1000):
    # 分解出个位、十位、百位
    digit1 = num // 100
    digit2 = (num // 10) % 10
    digit3 = num % 10
    # 如果满足水仙花数的条件，则输出
    if num == digit1 ** 3 + digit2 ** 3 + digit3 ** 3:
        print(num)

输出结果为：

153
370
371
407