定义一个函数fun(n),根据正整数参数n,返回其各位数字的平方和,并在主程序中调用。
时间: 2023-06-18 16:06:52 浏览: 323
以下是一个Python实现的函数fun(n):
```python
def fun(n):
"""
根据正整数参数n,返回其各位数字的平方和
"""
s = 0
while n > 0:
digit = n % 10
s += digit ** 2
n //= 10
return s
```
在主程序中,我们可以调用这个函数并输出结果:
```python
n = 123
print(fun(n)) # 输出14
```
以上程序会输出14,因为1^2 + 2^2 + 3^2 = 1 + 4 + 9 = 14。
相关问题
编写函数求1-n中奇数的平方和,用fun函数
好的,下面是使用 `functools.reduce()` 函数和 `lambda` 表达式编写的实现代码:
```python
import functools
def sum_of_odd_squares(n):
"""
求1-n中奇数的平方和
参数:
n -- 正整数,上界
返回值:
sum -- 整数,1-n中奇数的平方和
"""
odd_nums = filter(lambda x: x % 2 == 1, range(1, n+1))
sum = functools.reduce(lambda x, y: x + y**2, odd_nums, 0)
return sum
```
这个函数使用了 `filter()` 函数和 `lambda` 表达式来筛选出1到n中的奇数,然后使用 `functools.reduce()` 函数和 `lambda` 表达式来计算它们的平方和。你可以调用这个函数,比如:
```python
>>> sum_of_odd_squares(5)
35
```
这表示1到5中的奇数(1, 3, 5)的平方和为35。
阅读全文