以下是一个编写递归阶乘函数的代码示例,该函数接受用户输入并将其放入np数组中,并打印出它们的阶乘:
import numpy as np
def factorial(n):
if n == 0:
return 1
else:
return n * factorial(n-1)
# 获取用户输入
n = int(input("请输入一个非负整数:"))
# 创建一个np数组
arr = np.array([factorial(i) for i in range(n+1)])
# 打印出阶乘结果
print(arr)
该代码首先定义了一个递归函数factorial,用于计算阶乘。然后使用input
函数获取用户输入的一个非负整数n。接下来,使用列表推导式将从0到n的每个整数传递给factorial函数,并将结果存储在一个np数组arr中。最后,打印出数组arr的内容,即为用户输入的每个数的阶乘。