如何从函数中获取返回值并将其放在数据框列中

# 定义一个函数，计算输入参数的平方
calculate_square <- function(x) {
  return(x^2)
}

# 创建一个空的数据框
result_df <- data.frame()

# 创建一个输入参数的向量
input_vector <- c(1, 2, 3, 4, 5)

# 使用循环遍历数据框的每一行，调用函数并将返回值存储在数据框的相应列中
for (i in 1:length(input_vector)) {
  result <- calculate_square(input_vector[i])
  result_df <- rbind(result_df, result)
}

# 将结果列添加到数据框中
result_df <- cbind(input_vector, result_df)

# 打印结果数据框
print(result_df)