8

我有时需要根据其中一个变量的值从 data.frame 中提取特定行。具有最大值 ( ) 和最小值 ( )R的内置函数，可让我轻松提取这些行。which.max()which.min()

中位数是否有等价物？还是我最好的选择是编写自己的函数？

这是一个示例 data.frame 以及我将如何使用which.max()and which.min()：

set.seed(1) # so you can reproduce this example
dat = data.frame(V1 = 1:10, V2 = rnorm(10), V3 = rnorm(10), 
                 V4 = sample(1:20, 10, replace=T))

# To return the first row, which contains the max value in V4
dat[which.max(dat$V4), ]
# To return the seventh row, which contains the min value in V4
dat[which.min(dat$V4), ]

对于这个特定示例，由于存在偶数个观察值，我需要返回两行，在本例中为第 2 行和第 10 行。

更新

似乎没有为此的内置函数。因此，以Sacha 的回复为起点，我编写了这个函数：

which.median = function(x) {
  if (length(x) %% 2 != 0) {
    which(x == median(x))
  } else if (length(x) %% 2 == 0) {
    a = sort(x)[c(length(x)/2, length(x)/2+1)]
    c(which(x == a[1]), which(x == a[2]))
  }
}

我可以按如下方式使用它：

# make one data.frame with an odd number of rows
dat2 = dat[-10, ]
# Median rows from 'dat' (even number of rows) and 'dat2' (odd number of rows)
dat[which.median(dat$V4), ]
dat2[which.median(dat2$V4), ]

有什么建议可以改进吗？

4

6 回答 6

15

虽然 Sacha 的解决方案非常笼统，但中位数（或其他分位数）是顺序统计数据，因此您可以从order (x)（而不是sort (x)分位数值）计算相应的索引。

调查quantile，可以使用类型 1 或 3，在某些情况下，所有其他类型都会导致两个值的（加权）平均值。

quantile我选择了类型 3，并从线索中复制和粘贴一点：

which.quantile <- function (x, probs, na.rm = FALSE){
  if (! na.rm & any (is.na (x)))
  return (rep (NA_integer_, length (probs)))

  o <- order (x)
  n <- sum (! is.na (x))
  o <- o [seq_len (n)]

  nppm <- n * probs - 0.5
  j <- floor(nppm)
  h <- ifelse((nppm == j) & ((j%%2L) == 0L), 0, 1)
  j <- j + h

  j [j == 0] <- 1
  o[j]
}

一个小测试：

> x <-c (2.34, 5.83, NA, 9.34, 8.53, 6.42, NA, 8.07, NA, 0.77)
> probs <- c (0, .23, .5, .6, 1)
> which.quantile (x, probs, na.rm = TRUE)
[1] 10  1  6  6  4
> x [which.quantile (x, probs, na.rm = TRUE)] == quantile (x, probs, na.rm = TRUE, type = 3)

  0%  23%  50%  60% 100% 
TRUE TRUE TRUE TRUE TRUE

这是您的示例：

> dat [which.quantile (dat$V4, c (0, .5, 1)),]
  V1         V2          V3 V4
7  7  0.4874291 -0.01619026  1
2  2  0.1836433  0.38984324 13
1  1 -0.6264538  1.51178117 17
于 2012-04-21T10:14:09.310 回答
9

我认为只是：

which(dat$V4 == median(dat$V4))

但要小心，因为如果没有一个中间数字，中位数取两个数字的平均值。例如median(1:4)，给出不匹配任何元素的 2.5。

编辑

这是一个函数，它将为您提供中位数的元素或与中位数平均值的第一个匹配项，类似于如何which.min()为您提供仅等于最小值的第一个元素：

whichmedian <- function(x) which.min(abs(x - median(x)))

例如：

> whichmedian(1:4)
[1] 2
于 2012-04-21T05:48:20.550 回答
2

我编写了一个更全面的函数来满足我的需求：

row.extractor = function(data, extract.by, what) {
# data = your data.frame
# extract.by = the variable that you are extracting by, either
#              as its index number or by name
# what = either "min", "max", "median", or "all", with quotes
  if (is.numeric(extract.by) == 1) {
    extract.by = extract.by
  } else if (is.numeric(extract.by) != 0) {
    extract.by = which(colnames(dat) %in% "extract.by")
  } 
  which.median = function(data, extract.by) {
    a = data[, extract.by]
    if (length(a) %% 2 != 0) {
      which(a == median(a))
    } else if (length(a) %% 2 == 0) {
      b = sort(a)[c(length(a)/2, length(a)/2+1)]
      c(max(which(a == b[1])), min(which(a == b[2])))
    }
  }
  X1 = data[which(data[extract.by] == min(data[extract.by])), ] 
  X2 = data[which(data[extract.by] == max(data[extract.by])), ]
  X3 = data[which.median(data, extract.by), ]
  if (what == "min") {
    X1
  } else if (what == "max") {
    X2
  } else if (what == "median") {
    X3
  } else if (what == "all") {
    rbind(X1, X3, X2)
  }
}

一些示例用法：

> row.extractor(dat, "V4", "max")
  V1         V2       V3 V4
1  1 -0.6264538 1.511781 17
> row.extractor(dat, 4, "min")
  V1        V2          V3 V4
7  7 0.4874291 -0.01619026  1
> row.extractor(dat, "V4", "all")
   V1         V2          V3 V4
7   7  0.4874291 -0.01619026  1
2   2  0.1836433  0.38984324 13
10 10 -0.3053884  0.59390132 14
4   1 -0.6264538  1.51178117 17
于 2012-04-21T09:23:25.327 回答
2

假设要从中获取中位数的向量是x。

该函数which.min(x[x>=median(x)])将给出中值 iflength(x)=2*n+1或两个中间值中的较大值 if length(x)=2*n。如果您想获得两个中间值中较小的一个，您可以稍微调整它。

于 2015-10-25T09:39:42.880 回答
1

在 Sacha 和 cbeleites 给出的答案的基础上，这是一个获得包容性分位数指数的函数。与先前答案的一个不同之处在于，该type论点是公开的，并且会产生略有不同的分位数结果（请参阅 参考资料?quantile）。如果性能是一个问题，可以将 sapply 替换为parallel包中的版本 - 例如unlist(mclapply(...)).

# Extract indices corresponding to inclusive quantiles
# EXAMPLE:
#
#   x <- c(2.34, 5.83, NA, 9.34, 8.53, 6.42, NA, 8.07, NA, 0.77)
#   probs <- c(0, .23, .5, .6, 1)
#   which.quantile(x, probs, na.rm = TRUE)
#
# OUTPUT: 10  1  6  8  4
#
#   x[ which.quantile(x, probs, na.rm = TRUE) ]
#
# OUTPUT: 0.77 2.34 6.42 8.07 9.34
#
#   x <- c(2, 1, 3)
#   p <- c(0.5)
#   x[ which.quantile(x, p) ]
#
# OUTPUT: 2
which.quantile <- function (x,
                            probs,
                            na.rm = FALSE,
                            type = 7) {
  stopifnot(all(probs >= 0.0))
  stopifnot(all(probs <= 1.0))
  quants = quantile(x,
                    probs = probs,
                    na.rm = na.rm,
                    type = type)
  which.nearest <- function(quant) {
    return(which.min(abs(x - quant)))
  }
  return(sapply(X = quants, FUN = which.nearest))
}
于 2018-11-26T12:07:17.093 回答
0

我们只需要一个通过近似匹配返回值位置的函数：

match.approx <- function(x, y) {
    ## Purpose: Match Approximately for Numerical Data
    ## Arguments:
    ##   "x":  a vector of numeric values.
    ##   "y":  a vector of numeric values. 
    ## RETURN:
    ##   The index in "y" that indicates the closest y value to each of "x" value. 
    ## ________________________________________________
    
    sapply(x, function(x0) which.min(abs(x0 - y)))
}
if (F) {
  match.approx(c(4.2, 1.2, 15), 1:10)                #  4  1 10
}

以下是查找分位数位置的示例：

set.seed(1)
a <- rnorm(100)
match.approx(quantile(a), a)
# 0%  25%  50%  75% 100% 
# 14   29   23   63   61
于 2020-09-18T14:08:52.917 回答