在列包含 R 中的特定值的条件下，如何从具有多列的表中获取整个列

Question

我在 R 中有一个简单的表，其中包含 3 列（名为“countries_A”、“countries_B”和“countries_C”），每列包含 4 个国家/地区。 现在我想做的是编写一个 function 在表中搜索特定国家，比如“意大利”，然后返回该国家所在列的所有元素作为向量或列表（除了那个特定的国家）。 因此，在我的示例中，作为“countries_B”列中的“Italy”以及瑞典、西班牙和瑞士，这意味着我将获得一个包含“countries_B”列中除意大利以外的所有元素的单个向量，这意味着我将获得“[瑞典、西班牙和瑞士]”作为答案。 我真的很感激任何帮助。 如果可能的话，我希望尽可能使用像 dtplyr 这样的库来对搜索进行矢量化。 下面，我附上了我的表格的屏幕截图和 R 脚本以生成该表格。 提前谢谢了。

countries <- data.frame("countries_A" =
                          c('Belgium','Holland', 'France', 'Germany'), 
                      "countries_B" = c('Sweden','Italy','Spain','Switzerland'),
                     "countries_C"= c('England','Denmark','Portugal','Hungary'))

Answer 1

让我知道这是否对您有帮助：

find.country <- function(df, country){
      
      df$ID <- seq.int(nrow(df)) 
      
      df1 <-df %>%
        pivot_longer(cols = -ID) %>%
        filter(value == {country})
      
      column.name <- df1$name
      
      df2 <- df %>%
        pivot_longer(cols = -ID) %>%
        filter(name == column.name) %>%
        filter(value != {country})
      
      vector <- as.vector(df2$value)
      return(vector)
      
      
    }
    
 find.country(df = countries, country = "Italy")


    [1] "Sweden"      "Spain"       "Switzerland"

Answer 2

不太优雅：

get_vector = function(df_countries, country){
column = countries %>% mutate(across(everything(),
                                     ~if_else(country %in% .x, cur_column(), NULL))) %>%
  mutate(var = coalesce(countries_A, countries_B, countries_C)) %>%
  select(var) %>% slice(1) %>% pull
vector = countries %>% select(column) %>% filter(.data[[column]] != country) %>% pull %>% as.vector
return(vector)
}

get_vector(countries, 'Sweden')

output：

> get_vector(countries, 'Sweden')
[1] "Italy"       "Spain"       "Switzerland"