在数据框中将因子转换为数字 [英] convert factors to numeric in dataframe

问题描述

我有一个非常大的数据帧，包含2个级别的因子，级别为否和是。
我想将级别替换为数值，以使否变为0，是变为1。

我想应用一个适用于数据框架的功能。

一个简单的例子：

 > df 
abcd 
 1 1否是1 
 2 2否否3 
 3 3是否123 
 4 4是是12 
 5 5否是231 
 6 6否否21 
 7 7是否21 
 8 8是否21 
 
> str（df）
'data.frame'：8 obs。的4个变量：
 $ a：int 1 2 3 4 5 6 7 8 
 $ b：因子w / 2级别否，是：1 1 2 2 1 1 2 2 
 $ c：因素w / 2级别否，是：2 1 1 2 2 1 1 1 
 $ d：int 1 3 123 12 231 21 21 21 
  

想要的结果：

 > df 
abcd 
 1 1 0 1 1 
 2 2 0 0 3 
 3 3 1 0 123 
 4 4 1 1 12 
 5 5 0 1 231 
 6 6 0 0 21 
 7 7 1 0 21 
 
> str（df）
'data.frame'：8 obs。的4个变量：
 $ a：int 1 2 3 4 5 6 7 8 
 $ b：int 0 0 1 1 0 0 1 1 
 $ c：int 1 0 0 1 1 0 0 0 
 $ d：int 1 3 123 12 231 21 21 21 
  

解决方案

尝试

  df [2：3]<  -  lapply（df [2：3]函数（x）as.numeric（x）-1）
 df 
＃abcd 
＃1 1 0 1 1 
＃2 2 0 0 3 
＃3 3 1 0 123 
＃4 4 1 1 12 
＃5 5 0 1 231 
＃6 6 0 0 21 
＃7 7 1 0 21 
＃8 8 1 0 21 
  

这可以包装成一个函数

  f1<  -  function（dat）{
 indx<  -  sapply（dat，is.factor）
 dat [indx]如果（any（x％in％c（Yes，
No）））as.numeric（x）-1 else x）
 dat 
} 
 
 f1（df）
  

I have a very large data frame containing 2 levels of a factor, levels "No" and "Yes". I would like to replace the levels to numeric values, so that "No" turns into 0, and "Yes" turns into 1.

I would like to apply a function that works on the data frame.

A simple example to work on:

> df
  a   b   c   d
1 1  No Yes   1
2 2  No  No   3
3 3 Yes  No 123
4 4 Yes Yes  12
5 5  No Yes 231
6 6  No  No  21
7 7 Yes  No  21
8 8 Yes  No  21

> str(df)
'data.frame':   8 obs. of  4 variables:
 $ a: int  1 2 3 4 5 6 7 8
 $ b: Factor w/ 2 levels "No","Yes": 1 1 2 2 1 1 2 2
 $ c: Factor w/ 2 levels "No","Yes": 2 1 1 2 2 1 1 1
 $ d: int  1 3 123 12 231 21 21 21

Wanted results :

> df
  a b c   d
1 1 0 1   1
2 2 0 0   3
3 3 1 0 123
4 4 1 1  12
5 5 0 1 231
6 6 0 0  21
7 7 1 0  21

> str(df)
'data.frame':   8 obs. of  4 variables:
 $ a: int  1 2 3 4 5 6 7 8
 $ b: int  0 0 1 1 0 0 1 1
 $ c: int  1 0 0 1 1 0 0 0
 $ d: int  1 3 123 12 231 21 21 21

解决方案

Try

df[2:3] <- lapply(df[2:3], function(x) as.numeric(x)-1)
df
#   a b c   d
#1 1 0 1   1
#2 2 0 0   3
#3 3 1 0 123
#4 4 1 1  12
#5 5 0 1 231
#6 6 0 0  21
#7 7 1 0  21
#8 8 1 0  21

This could be wrapped into a function

f1 <- function(dat){
 indx <- sapply(dat, is.factor)
 dat[indx] <- lapply(dat[indx], function(x) if(any(x %in% c("Yes",
             "No"))) as.numeric(x)-1 else x)
 dat 
}

f1(df)

这篇关于在数据框中将因子转换为数字的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！