如何在循环中调用函数来计算 R 中的加权交叉表答案

【问题标题】：How to call a function in a loop to compute weighted crosstables in R如何在循环中调用函数来计算 R 中的加权交叉表
【发布时间】：2020-09-30 14:47:51
【问题描述】：

我想计算变量x 和同一dataframe 的几个y 变量之间的几个加权交叉表。我创建了一个函数来执行此操作，但我无法循环我的 y 变量列表。

我给出了一个使用mpg 数据的过程示例：这里，我的y 变量是cyl 和year。 x 是 class。

library(questionr)

# include some weights in mpg2 (same weight)

mpg2 <- mpg %>%
  mutate(weight = 1/234)
#> Error in mpg %>% mutate(weight = 1/234): could not find function "%>%"

# crosstab class by cyl using weights
questionr::wtd.table(mpg2$class, mpg2$cyl,
                                  weights=mpg2$weight,
                                  digits = 0, na.show=FALSE)
#> Error in questionr::wtd.table(mpg2$class, mpg2$cyl, weights = mpg2$weight, : object 'mpg2' not found

# using a function
calc_table <- function(x, y) {
  return(questionr::wtd.table(x, y,
                       weights=mpg2$weight,
                       digits = 0, na.show=FALSE))
}

# No problem when describing the variable as mpg2$cyl
calc_table(mpg2$class, mpg2$cyl)
#> Error in questionr::wtd.table(x, y, weights = mpg2$weight, digits = 0, : object 'mpg2' not found

# But I want to crosstab class with cyl (mpg2[5])
# and also class with year (mpg2[4]) 
# referencing the variable in a loop
# My main goal is to use the function for several variables

for (i in c(4:5)){
  calc_table(mpg2$class, mpg2[i])
}
#> Error in questionr::wtd.table(x, y, weights = mpg2$weight, digits = 0, : object 'mpg2' not found

【问题讨论】：

标签： r function weighted

【解决方案1】：

问题是您使用了mpg2[i]，它返回一个数据帧或tibble，而不是mpg2[[i]]，它返回一个向量。但是，除了这个问题，我建议循环列名而不是位置。我还添加了一个列表来保存结果。试试这个：

library(dplyr)
library(questionr)

# include some weights in mpg2 (same weight)

mpg2 <- ggplot2::mpg %>%
  mutate(weight = 1/234)

# using a function
calc_table <- function(x, y) {
  questionr::wtd.table(x, y, weights=mpg2$weight, digits = 0, na.show=FALSE)
}

table_list <- list()
for (i in c("cyl", "year")){
  print(table_list[[i]] <- calc_table(mpg2$class, mpg2[[i]]))
}
#>                      4           5           6           8
#> 2seater    0.000000000 0.000000000 0.000000000 0.021367521
#> compact    0.136752137 0.008547009 0.055555556 0.000000000
#> midsize    0.068376068 0.000000000 0.098290598 0.008547009
#> minivan    0.004273504 0.000000000 0.042735043 0.000000000
#> pickup     0.012820513 0.000000000 0.042735043 0.085470085
#> subcompact 0.089743590 0.008547009 0.029914530 0.021367521
#> suv        0.034188034 0.000000000 0.068376068 0.162393162
#>                   1999        2008
#> 2seater    0.008547009 0.012820513
#> compact    0.106837607 0.094017094
#> midsize    0.085470085 0.089743590
#> minivan    0.025641026 0.021367521
#> pickup     0.068376068 0.072649573
#> subcompact 0.081196581 0.068376068
#> suv        0.123931624 0.141025641

【讨论】：