下面是一个用Python编写的函数,用于计算向量的前k个元素的平方:
def calc_square(vector, k):
result = []
for i in range(k):
result.append(vector[i] ** 2)
return result
使用示例:
vector = [1, 2, 3, 4, 5]
k = 3
result = calc_square(vector, k)
print(result) # 输出: [1, 4, 9]
在这个示例中,我们定义了一个名为calc_square
的函数,它接受两个参数:vector
是输入的向量,k
是要计算平方的元素数量。函数返回一个包含前k个元素平方的列表。
在函数内部,我们使用一个循环迭代前k个元素,并使用指数运算符**
计算每个元素的平方。然后,我们将计算结果添加到result
列表中,并在循环结束后返回该列表。