如何编写函数来更改字符串并进行算术运算

Question

我在数据框中有一个变量，其中包含的值要么是 Varies with device，要么是 a number with M，要么是 a number with k。

例如：

10M
1.2M
120k
Varies with device

我想使用the values with M 并简单地删除M，我想使用the values with k 并删除k 并将它们乘以0.001。

示例输出：

10
1.2
0.12
Varies with device

我希望输出在dataframe 中的一个变量中，或者替换原始变量，或者作为一个新变量。

我试图创建一个这样的函数，但我无法让它工作。

convert <- function(x) {
  if(grep("M$", x)) {
  str_sub(x, 1, -1)
  as.numeric(x)
  } else if(grep("k$", x)) {
    str_sub(x, 1, -1)
    as.numeric(x)
    (x*0.001)
  }
}

Answer 1

test <- c("10M 1.2M 120k Varies with device")

library(stringr)
library(magrittr)

str_replace_all(test,"[0-9\\.]+(?=k)",function(x){as.numeric(x)/1000}) %>%
  gsub("k|M","",.)

[1] "10 1.2 0.12 Varies with device"

函数str_replace_all 将匹配的模式替换为其他模式，或由函数转换的匹配模式。我在这里用它来除以每 1000 个数字 [0-9\\.]+ 后跟 k (?=k)。

然后我用 gsub 删除 k 和 M（我匹配 k 或 M "k|M" 并将其替换为空）

它也适用于矢量

test <- c("10M","1.2M", "120k"," Varies with device")
str_replace_all(test,"[0-9\\.]+(?=k)",function(x){as.numeric(x)/1000}) %>%
   gsub("k|M","",.)

[1] "10"                  "1.2"                 "0.12"                " Varies with device"

Answer 2

我对您的功能做了一些调整。它现在应该可以正常工作了。

convert <- function(x) {
  if(length(grep("M$", x)) > 0) {
    x <- str_sub(x, 1, nchar(x)-1)
    as.numeric(x)
  } else if(length(grep("k$", x)) > 0) {
    x <- str_sub(x, 1, nchar(x)-1)
    as.numeric(x)*0.001
  } else {
    x
  }

}