2014-12-17 4 views
0

Я хотел бы создать пользовательскую функцию с циклом for, которая добавляет к каждому столбцу имя столбца, чтобы добавить ясность в мой фрейм данных.пользовательская функция и для цикла

Вот код (но не работает):

addncol = function(df) 
    { 
    for(i in 1:ncol(df)) 
     { 
     names(df)[[i]] <- paste(names(df)[[i]], i, sep = '_') 
     } 
    } 

одна для контура работает хорошо, но в функции она не работает (у меня нет, хотя сообщения об ошибке).

+0

имена (df) - это не список, а вектор. Было бы более типично использовать его таким образом: .......... 'names (df) [i] <- ...'. Но это не основная ошибка, тогда вам также нужно вернуть ее и вернуть ее к исходному 'df', поскольку то, что вы только что сделали, произошло только внутри функции. Или вы могли бы предположить использование '<< -' –

+0

Спасибо Свену. Есть ли способ назначить его обратно df автоматически внутри функции? Вместо того, чтобы вводить: 'DF <- addncol (DF)' Вне функции, я только хочу, чтобы напечатать: 'addncol (DF)' И моя новая ДФ модифицируется, не показан в console – Synleb

+0

Не знаете, кто такой Свен, но я уже упоминал '<< -'. Это считается мошенничеством cognoscenti. –

ответ

2

Функция не возвращает фрейм данных df. Он будет работать, если вы добавите df в качестве последней строки вашей функции.

Однако более простой вариантом без петель является Векторизованным подходом с setNames:

addncol <- function(df) 
    setNames(df, paste(names(df), seq(df), sep = "_")) 
+0

'setNames' действительно намного лучший выбор здесь. – MrFlick

1

Вы можете использовать data.tables для этого, каких обновлений по ссылке с помощью setnames(...) (обратите внимание на другое написание - это другая функция).

library(data.table) 
DT <- as.data.table(mtcars) # example only 
head(DT) 
mpg cyl disp hp drat wt qsec vs am gear carb 
# 1: 21.0 6 160 110 3.90 2.620 16.46 0 1 4 4 
# 2: 21.0 6 160 110 3.90 2.875 17.02 0 1 4 4 
# 3: 22.8 4 108 93 3.85 2.320 18.61 1 1 4 1 
# 4: 21.4 6 258 110 3.08 3.215 19.44 1 0 3 1 
# 5: 18.7 8 360 175 3.15 3.440 17.02 0 0 3 2 
# 6: 18.1 6 225 105 2.76 3.460 20.22 1 0 3 1 
addncol <- function(DT) setnames(DT,paste(names(DT),seq(DT),sep="_")) 
addncol(DT) 
head(DT) 
# mpg_1 cyl_2 disp_3 hp_4 drat_5 wt_6 qsec_7 vs_8 am_9 gear_10 carb_11 
# 1: 21.0  6 160 110 3.90 2.620 16.46 0 1  4  4 
# 2: 21.0  6 160 110 3.90 2.875 17.02 0 1  4  4 
# 3: 22.8  4 108 93 3.85 2.320 18.61 1 1  4  1 
# 4: 21.4  6 258 110 3.08 3.215 19.44 1 0  3  1 
# 5: 18.7  8 360 175 3.15 3.440 17.02 0 0  3  2 
# 6: 18.1  6 225 105 2.76 3.460 20.22 1 0  3  1 
Смежные вопросы