在 tibble 中查找行最小值和列索引

Question

我有以下问题：

> df <- tibble(
     ID = LETTERS[1:4],
     a  = c(1,5,9,8),
     b  = c(5,9,8,2),
     c  = c(5,4,5,5)
)

> df
# A tibble: 4 x 4
  ID        a     b     c
  <chr> <dbl> <dbl> <dbl>
1 A         1     5     5
2 B         5     9     4
3 C         9     8     5
4 D         8     2     5
>

我想要的是获取列的行最小值 a:c 以及该最小值的列索引。
输出表应如下所示：

# A tibble: 4 x 6
  ID        a     b     c   Min Col_Index
  <chr> <dbl> <dbl> <dbl> <dbl>     <dbl>
1 A         1     5     5     1         1
2 B         5     9     4     4         3
3 C         9     8     5     5         3
4 D         8     2     5     2         2

我不想使用 rowwise()!

谢谢！

Answer 1

您可以将 pmin 与 do.call 一起使用以获得行最小值，并将值取反以与 max.col 一起使用以获得最小列索引。

library(dplyr)
library(purrr)

df %>%
  mutate(Min = do.call(pmin, select(., a:c)), 
         Col_Index = max.col(-select(., a:c)))

#  ID        a     b     c   Min Col_Index
#  <chr> <dbl> <dbl> <dbl> <dbl>     <int>
#1 A         1     5     5     1         1
#2 B         5     9     4     4         3
#3 C         9     8     5     5         3
#4 D         8     2     5     2         2

使用 purrr 的 pmap_dbl :

df %>%
  mutate(Min = pmap_dbl(select(., a:c), ~min(c(...))),
         Col_Index = pmap_dbl(select(., a:c), ~which.min(c(...))))

Answer 2

一个选项可以是：

df %>%
 rowwise() %>%
 mutate(min = min(c_across(a:c)),
        min_index = which.min(c_across(a:c)))

  ID        a     b     c   min min_index
  <chr> <dbl> <dbl> <dbl> <dbl>     <int>
1 A         1     5     5     1         1
2 B         5     9     4     4         3
3 C         9     8     5     5         3
4 D         8     2     5     2         2

Answer 3

基础 R 解决方案：

setNames(cbind(df, t(apply(df[, vapply(df, is.numeric, logical(1))], 1, function(row) {
  cbind(min(row), which.min(row))}))), c(names(df), "min", "col_index"))

在 tibble 中查找行最小值和列索引

Finding rowwise minimum and column index in a tibble

r

purrr

tibble