r - 向量化简单格式化函数以处理带有 NA 的向量

Question

我有一个简单的函数，用于在图表标签中以所需方式格式化货币:

curr_fmt <- function(x) {
    f <- function(x) {
        if (is.numeric(x)) {
            if (x >= 1e4) {
                x <- x / 1e3
                paste0(intToUtf8(163), scales::comma(x), "K")
            } else {
                paste0(intToUtf8(163), scales::comma(x))
            }
        } else {
            x
        }
    }

    vf <- Vectorize(f, "x")
    vf(x)

}

例子

当应用于数值向量时，它会产生预期的结果:

>> curr_fmt(c(1, 2e4, 3e4, 100))
[1] "£1"   "£20K" "£30K" "£100"

问题

如果向量包含 NA 值，则不会产生预期的结果:

>> curr_fmt(c(1, 2e4, 3e4, 100, NA))

Rerun with Debug Error in if (x >= 10000) { : missing value where TRUE/FALSE needed

问题必须与向量化有关，因为函数 f 以所需方式处理 NA 值:

>> f(NA)
[1] NA
>> f(3e4)
[1] "£30K"

期望的结果

curr_fmt(c(1, 2e4, 3e4, 100, NA))
[1] "£1"   "£20K" "£30K" "£100" NA

• 返回的对象应该是一个字符向量，其长度对应于传递的 x 向量的长度，带有 NA 值返回为 NA，货币值的格式与示例相同。

---

注意事项:

• 请注意，我对for-loop 解决方案不感兴趣
• mapply/sapply基于解决方案将被接受
• 理想情况下，我想看看如何正确使用 Vectorize在这种情况下
• 函数不必处理字符串

Answer 1

我们需要更改 if 条件，以便仅使用非 NA 值

curr_fmt <- function(x) {
    f <- function(x) {
        if (is.numeric(x) & !is.na(x)) { # changed here
            if (x >= 1e4) {
                x <- x / 1e3
                paste0(intToUtf8(163), scales::comma(x), "K")
            } else {
                paste0(intToUtf8(163), scales::comma(x))
            }
        } else {
            x
        }
    }

    vf <- Vectorize(f, "x")
    vf(x)

}

curr_fmt(c(1, 2e4, 3e4, 100, NA))
#[1] "£1"   "£20K" "£30K" "£100" NA    
curr_fmt(c(1, 2e4, 3e4, 100))
#[1] "£1"   "£20K" "£30K" "£100"