数据框中的输出回路Kolmogorov Smirnov测试(ks.test)

时间:2019-11-27 15:27:00

标签: r loops dataframe kolmogorov-smirnov

我想将循环中记录的ks.test的输出合并到数据帧或文件中,而不是在控制台中打印1155测试的输出... :-)。

column_equality_stats = function(data, lab_stats1, lab_stats2, min_count=100) {
  for(i in 1:length(lab_stats1)) {
    lab_testcodes_1 = lab_stats1[i]
    lab_testcodes_2 = lab_stats2[i]
    equal_columns <- filter(data, lab_testcode==lab_testcodes_1 | lab_testcode==lab_testcodes_2)
    col1 <- equal_columns[equal_columns$lab_testcode==lab_testcodes_1, 'lab_result']
    col2 <- equal_columns[equal_columns$lab_testcode==lab_testcodes_2, 'lab_result']
    if(sum(!is.na(col1))>min_count && sum(!is.na(col2))>min_count){
      stats <- ks.test(col1, col2)
      print(stats)

    }
  }
}

我希望有一个data.frame,其中包含以下列:col1和col2的名称(方程值),p值和D值。

Utopian data frame

非常感谢您!

1 个答案:

答案 0 :(得分:0)

这是我所能做到的。请编辑您的问题,并添加一个可复制的示例。

我需要知道如何定义您的数据。我无法运行您的代码! :-)

无论如何,只需为循环的每次运行创建一个数据框并将其绑定在一起。

purrr软件包的

map_dfr为您做到了。

library(purrr)

.column_equality_stats <- function(i, data, lab_stats1, lab_stats2, min_count = 100){

  lab_testcodes_1 <- lab_stats1[i]
  lab_testcodes_2 <- lab_stats2[i]
  equal_columns <- filter(data, lab_testcode==lab_testcodes_1 | lab_testcode==lab_testcodes_2)
  col1 <- equal_columns[equal_columns$lab_testcode==lab_testcodes_1, 'lab_result']
  col2 <- equal_columns[equal_columns$lab_testcode==lab_testcodes_2, 'lab_result']

  if(sum(!is.na(col1))>min_count && sum(!is.na(col2))>min_count){

    stats <- ks.test(col1, col2)
    res <- data.frame(col1 = lab_testcodes_1,
                      col2 = lab_testcodes_2,,
                      pvalue = stats$p.value,
                      dvalue = stats$statistics)

  } else {res <- data.frame()}

  res

}

column_equality_stats <- function(data, lab_stats1, lab_stats2, min_count=100) {

  map_dfr(seq_along(lab_stats1), 
          .column_equality_stats,
          data       = data,
          lab_stats1 = lab_stats1,
          lab_stats2 = lab_stats2,
          min_count  = min_count)

}