Как узнать, состоит ли вектор из одинаковых элементов? - PullRequest
12 голосов
/ 01 марта 2020

Как я могу проверить, имеет ли вектор все одинаковые элементы?

Например, скажем, у меня есть:

vec1 = rep(10,20)
vec2 = seq(1:20)

Как я могу показать, что vec1 имеет все одинаковые элементы?

Ответы [ 6 ]

12 голосов
/ 01 марта 2020

Опция diff.

diff(vec1)

Если элементы равны, их разность равна нулю.

all(diff(vec1) == 0)
#[1] TRUE

Или сравнить вектор с его первым элементом.

all(vec1 == vec1[1])
#[1] TRUE

Редактировать.

Было опубликовано несколько способов определения того, равны ли все элементы вектора, см. RHertel , Юрий Сарайкин , tmfmnk . Вот сравнительные тесты.

library(microbenchmark)
library(ggplot2)

f <- function(n){
  x <- rep(10, n)
  mb <- microbenchmark(
    var = var(x) == 0,
    sd = sd(x) == 0,
    diff = all(diff(x) == 0),
    extract = all(x == x[1]),
    unique = length(unique(x)) == 1
  )
  mb
}

sizes <- c(10, 100, seq(1e3, 1e4, by = 1e3))
mb_list <- lapply(sizes, f)
names(mb_list) <- sizes

res <- lapply(seq_along(mb_list), function(i){
  agg <- aggregate(time ~ expr, mb_list[[i]], median)
  agg$size <- sizes[i]
  agg
})
res <- do.call(rbind, res)

ggplot(res, aes(size, time, colour = expr)) +
  geom_point() +
  geom_line()

enter image description here

7 голосов
/ 01 марта 2020

Используйте дисперсию. Если все элементы вектора равны, дисперсия равна нулю:

allElementsEqual <- function(x) {!var(x)}

#allElementsEqual(vec1)
#[1] TRUE
#allElementsEqual(vec2)
#[1] FALSE
4 голосов
/ 01 марта 2020

количество уникальных значений

length(unique(vec1))
2 голосов
/ 01 марта 2020

Мы можем использовать n_distinct от dplyr

library(dplyr)
n_distinct(vec1)
1 голос
/ 01 марта 2020

Вы можете преобразовать его в фактор и получить уровни, которые должны быть только 1, если они все одинаковые.

vec1 <- (10,20)

vec <- factor(vec1)

levels(vec)
1 голос
/ 01 марта 2020

Вы также можете рассчитать стандартное отклонение и оценить, является ли оно нулевым:

sd(vec1) == 0

[1] TRUE

Или используя table():

length(table(vec1)) == 1

Или используя rle():

length(rle(vec1)$lengths) == 1
...