编写一个函数selfidem,提供一个大于1的正整数参数n,返回n位数中符合自幂数要求的数的列表。n位数的自幂数是指n位数中每位数的n次方的和等于该数自身,水仙花数就是3位自幂数。例如,利用该函数执行下面命令可以找出所有4位自幂数(即玫瑰花数)。 print(selfidem(4))
时间: 2023-06-16 22:04:47 浏览: 173
好的,以下是实现该功能的Python函数selfidem:
```python
def selfidem(n):
result = []
for i in range(10**(n-1), 10**n):
num = str(i)
sum = 0
for digit in num:
sum += int(digit)**n
if sum == i:
result.append(i)
return result
```
这个函数首先生成所有n位数,然后对于每个数,将其每位数的n次方加起来,判断是否等于该数本身,如果是,则加入结果列表。最后返回结果列表。
你可以使用以下方式调用该函数找出所有4位自幂数:
```python
print(selfidem(4))
```
输出结果为:
```
[1634, 8208, 9474]
```
这些数都是符合自幂数要求的四位数。
阅读全文