以下是一个简单的函数来计算向量中的1的个数的代码示例:
def count_ones(vector):
count = 0
for element in vector:
if element == 1:
count += 1
return count
# 调用示例
vector = [1, 0, 1, 1, 0, 1]
result = count_ones(vector)
print(result) # 输出:4
在这个例子中,我们定义了一个名为count_ones
的函数,它接受一个向量作为参数。在函数内部,我们使用一个for
循环遍历向量中的每个元素。如果元素的值等于1,我们将count
变量的值加1。最后,函数返回count
变量的值,即向量中1的个数。
在调用示例中,我们创建了一个向量vector
,包含了一些0和1的元素。我们调用count_ones
函数并将vector
作为参数传递给它。最后,我们打印出函数的返回值,即向量中1的个数。