Блог

ПОСТАНОВКА ЗАДАЧИ

У меня есть таблица, в которой некоторые столбцы начинаются со строки, которую я хотел бы удалить. Я пытался сделать это, используя новую функциональность «поперек» из dplyr и функции из stringr для манипулирования строками, но у меня ничего не вышло (вот почему я здесь!).).

ПРИМЕР КОДА

 library(tidyverse)

# Generate a short mock-up table
set.seed(1)
mockup <- tibble(n_col1 = sample(1:10, 5, FALSE),
                 v_col2 = sample(letters, 5, FALSE),
                 col3 = sample(10:20, 5, TRUE),
                 col4 = sample(LETTERS, 5, FALSE)) # More columns beginning with "n_", "v_" or nothing

# Remove the "n_" or "v_" strings from the column names
# This is as far as my skills go...
mockup <- mockup %>% mutate(across(starts_with(c("n_", "v_")), str_remove))
  

ВОПРОС

Есть идеи, как это сделать, по-моему или по-другому, коротким и эффективным способом?

across используется для манипулирования значениями в столбце, а не именами столбцов. Вы могли бы использовать new rename_with для переименования столбцов.

 library(dplyr)
mockup %>%  rename_with(~sub('^(n|v)_', '', .))
#rename_all in old dplyr
#mockup %>%  rename_all(~sub('^(n|v)_', '', .))

#   col1 col2   col3 col4 
#  <int> <chr> <int> <chr>
#1     9 w        10 N    
#2     4 k        14 J    
#3     7 n        14 G    
#4     1 r        19 I    
#5     2 s        15 O    
  

Или выполнение этого в базовом R :

 names(mockup) <- sub('^(n|v)_', '', names(mockup))
  

Мы можем использовать str_remove с dplyr

 library(dplyr)
library(stringr)
mockup %>%
    rename_with(~ str_remove(., '^[nv]_'))
# A tibble: 5 x 4
#   col1 col2   col3 col4 
#  <int> <chr> <int> <chr>
#1     9 w        10 N    
#2     4 k        14 J    
#3     7 n        14 G    
#4     1 r        19 I    
#5     2 s        15 O