Python 判断一个数字是否为 Armstrong 数

Document 对象参考手册 Python3 实例

Armstrong 数(也称为自幂数)是指一个 n 位数,它的每个位上的数字的 n 次幂之和等于它本身。例如,153 是一个 Armstrong 数,因为 1^3 + 5^3 + 3^3 = 153。

下面是一个 Python 程序,用于判断一个数字是否为 Armstrong 数。

实例

def is_armstrong_number(num):
    # 将数字转换为字符串,以便逐个获取每个数字
    num_str = str(num)
    # 获取数字的位数
    n = len(num_str)
    # 计算每个数字的 n 次幂之和
    sum_of_powers = sum(int(digit) ** n for digit in num_str)
    # 判断和是否等于原数字
    return sum_of_powers == num

# 测试
number = 153
if is_armstrong_number(number):
    print(f"{number} 是一个 Armstrong 数")
else:
    print(f"{number} 不是一个 Armstrong 数")

代码解析:

  1. num_str = str(num):将输入的数字转换为字符串,以便逐个获取每个数字。
  2. n = len(num_str):获取数字的位数。
  3. sum_of_powers = sum(int(digit) ** n for digit in num_str):计算每个数字的 n 次幂之和。这里使用了列表推导式来遍历每个数字,并将其转换为整数后进行幂运算,最后使用 sum() 函数求和。
  4. return sum_of_powers == num:判断计算得到的和是否等于原数字,如果相等则返回 True,否则返回 False

输出结果:

153 是一个 Armstrong 数

Document 对象参考手册 Python3 实例