Динамічно вибирайте стовпці кадру даних, використовуючи $ та значення символу


122

У мене є вектор різних назв стовпців, і я хочу мати можливість перебирати петлі над кожним із них, щоб отримати цей стовпець із data.frame. Наприклад, розглянемо набір даних mtcarsі деякі імена змінних, що зберігаються у символьному векторі cols. Коли я намагаюся вибрати змінну з mtcarsвикористанням динамічного підмножини cols, яка є нижчою частиною цих робіт

cols <- c("mpg", "cyl", "am")
col <- cols[1]
col
# [1] "mpg"

mtcars$col
# NULL
mtcars$cols[1]
# NULL

як змусити їх повернути ті самі значення, що і

mtcars$mpg

Крім того, як я можу перевести цикл на всі стовпці, colsщоб отримати значення в якомусь циклі.

for(x in seq_along(cols)) {
   value <- mtcars[ order(mtcars$cols[x]), ]
}

Відповіді:


183

Ви не можете робити такий тип підмножини $. У вихідному коді ( R/src/main/subset.c) зазначено:

/ * Оператор підмножини $.
Нам потрібно бути впевненим лише для оцінки першого аргументу.
Другий буде символом, який потрібно узгоджувати, а не оцінювати.
* /

Другий аргумент? Що?! Ви повинні розуміти , що $, як і все інше в R, ( в тому числі, наприклад (, +, і ^т.д.) є функцією, яка приймає аргументи і оцінюється. df$V1може бути переписаний як

`$`(df , V1)

чи справді

`$`(df , "V1")

Але ...

`$`(df , paste0("V1") )

... наприклад, ніколи не вийде, і нічого іншого, що потрібно спочатку оцінити у другому аргументі, не буде. Ви можете передавати лише рядок, який ніколи не буває оцінюється.

Натомість використовуйте [(або [[якщо ви хочете витягти лише один стовпець як вектор).

Наприклад,

var <- "mpg"
#Doesn't work
mtcars$var
#These both work, but note that what they return is different
# the first is a vector, the second is a data.frame
mtcars[[var]]
mtcars[var]

Ви можете виконати замовлення без циклів, використовуючи do.callдля побудови виклику до order. Ось відтворюваний приклад нижче:

#  set seed for reproducibility
set.seed(123)
df <- data.frame( col1 = sample(5,10,repl=T) , col2 = sample(5,10,repl=T) , col3 = sample(5,10,repl=T) )

#  We want to sort by 'col3' then by 'col1'
sort_list <- c("col3","col1")

#  Use 'do.call' to call order. Seccond argument in do.call is a list of arguments
#  to pass to the first argument, in this case 'order'.
#  Since  a data.frame is really a list, we just subset the data.frame
#  according to the columns we want to sort in, in that order
df[ do.call( order , df[ , match( sort_list , names(df) ) ]  ) , ]

   col1 col2 col3
10    3    5    1
9     3    2    2
7     3    2    3
8     5    1    3
6     1    5    4
3     3    4    4
2     4    3    4
5     5    1    4
1     2    5    5
4     5    3    5

Чи змінилася ця ситуація за останні роки?
Дунуа

4

Якщо я правильно розумію, у вас є вектор, що містить імена змінних, і хотілося б переглядати кожне ім’я та сортувати їхній кадр даних за ними. Якщо так, цей приклад повинен ілюструвати рішення для вас. Основна проблема у вашому (повний приклад не завершений, тому я "не впевнений, що ще вам може бути пропущено") полягає в тому, що він повинен бути order(Q1_R1000[,parameter[X]])замість цього order(Q1_R1000$parameter[X]), оскільки параметр - це зовнішній об'єкт, який містить ім'я змінної, протилежне прямому стовпцю вашого фрейму даних (що коли це $буде доречно).

set.seed(1)
dat <- data.frame(var1=round(rnorm(10)),
                   var2=round(rnorm(10)),
                   var3=round(rnorm(10)))
param <- paste0("var",1:3)
dat
#   var1 var2 var3
#1    -1    2    1
#2     0    0    1
#3    -1   -1    0
#4     2   -2   -2
#5     0    1    1
#6    -1    0    0
#7     0    0    0
#8     1    1   -1
#9     1    1    0
#10    0    1    0

for(p in rev(param)){
   dat <- dat[order(dat[,p]),]
 }
dat
#   var1 var2 var3
#3    -1   -1    0
#6    -1    0    0
#1    -1    2    1
#7     0    0    0
#2     0    0    1
#10    0    1    0
#5     0    1    1
#8     1    1   -1
#9     1    1    0
#4     2   -2   -2

4

Використання dplyr забезпечує простий синтаксис для сортування кадрів даних

library(dplyr)
mtcars %>% arrange(gear, desc(mpg))

Можливо, буде корисно використовувати версію NSE, як показано тут, щоб дозволити динамічно будувати список сортування

sort_list <- c("gear", "desc(mpg)")
mtcars %>% arrange_(.dots = sort_list)

Що тут означає NSE?
дисципліна

1
@ нестандартне оцінювання; це для роботи із затримкою виразів для динамічного побудови коду з рядків замість жорсткого кодування. Дивіться тут для отримання додаткової інформації: cran.r-project.org/web/packages/lazyeval/vignettes/…
manotheshark

1

Ще одне рішення - використовувати #get:

> cols <- c("cyl", "am")
> get(cols[1], mtcars)
 [1] 6 6 4 6 8 6 8 4 4 6 6 8 8 8 8 8 8 4 4 4 4 8 8 8 8 4 4 4 8 6 8 4

0

Виникла подібна проблема через деякі файли CSV, які мали різні імена для одного стовпця.
Це було рішення:

Я написав функцію, щоб повернути перше дійсне ім'я стовпця у списку, а потім використав це ...

# Return the string name of the first name in names that is a column name in tbl
# else null
ChooseCorrectColumnName <- function(tbl, names) {
for(n in names) {
    if (n %in% colnames(tbl)) {
        return(n)
    }
}
return(null)
}

then...

cptcodefieldname = ChooseCorrectColumnName(file, c("CPT", "CPT.Code"))
icdcodefieldname = ChooseCorrectColumnName(file, c("ICD.10.CM.Code", "ICD10.Code"))

if (is.null(cptcodefieldname) || is.null(icdcodefieldname)) {
        print("Bad file column name")
}

# Here we use the hash table implementation where 
# we have a string key and list value so we need actual strings,
# not Factors
file[cptcodefieldname] = as.character(file[cptcodefieldname])
file[icdcodefieldname] = as.character(file[icdcodefieldname])
for (i in 1:length(file[cptcodefieldname])) {
    cpt_valid_icds[file[cptcodefieldname][i]] <<- unique(c(cpt_valid_icds[[file[cptcodefieldname][i]]], file[icdcodefieldname][i]))
}

0

якщо ви хочете вибрати стовпчик із конкретною назвою, тоді просто зробіть

A=mtcars[,which(conames(mtcars)==cols[1])]
#and then
colnames(mtcars)[A]=cols[1]

ви можете запустити його в циклі, а також зворотним способом додати динамічне ім'я, наприклад, якщо A - кадр даних, а xyz - стовпець, який повинен бути названий як x, то мені це подобається

A$tmp=xyz
colnames(A)[colnames(A)=="tmp"]=x

знову це також можна додати в циклі


Я не знаю, чому негативно проголосували, але це працює і простий спосіб замість того, щоб писати складні функції
makarand kulkarni


-1

занадто пізно .. але, мабуть, у мене є відповідь -

Ось мій зразок data.df dataframe -

   >study.df
   study   sample       collection_dt other_column
   1 DS-111 ES768098 2019-01-21:04:00:30         <NA>
   2 DS-111 ES768099 2018-12-20:08:00:30   some_value
   3 DS-111 ES768100                <NA>   some_value

І потім -

> ## Selecting Columns in an Given order
> ## Create ColNames vector as per your Preference
> 
> selectCols <- c('study','collection_dt','sample')
> 
> ## Select data from Study.df with help of selection vector
> selectCols %>% select(.data=study.df,.)
   study       collection_dt   sample
1 DS-111 2019-01-21:04:00:30 ES768098
2 DS-111 2018-12-20:08:00:30 ES768099
3 DS-111                <NA> ES768100
> 
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.