ChatGPT解决这个技术问题 Extra ChatGPT

提取向量的每个第 n 个元素

我想创建一个向量,其中每个元素都是另一个向量的 i+6th 元素。

例如,在长度为 120 的向量中,我想创建另一个长度为 20 的向量,其中每个元素都是初始向量的值 i, i+6, i+12, i+18...,即我想提取原始向量的每 6 个元素。


n
nico
a <- 1:120
b <- a[seq(1, length(a), 6)]

最好使用 seq.int(1L, length(a), 6L),至少对于长向量
@WojciechSobala 你能评论一下为什么它更好吗?
@DavidPell seq.int 在微基准测试中更快,但我怀疑实际程序中的任何性能提升都会与其他部分的运行时间相形见绌。
我讨厌将 Python 与 R 进行比较,但 PyRon 能有多好? a = 1:120; b = [::6]。 Python 做不到前者,R 做不到后者。
z
zx8754

获得连续片段的另一个技巧(除了已经提到的 seq 解决方案)是使用短逻辑向量并使用向量循环:

foo[ c( rep(FALSE, 5), TRUE ) ]

这种方法的一个优点是可以临时使用;为了使用 seq,您必须能够在向量上调用 lengthletters[letters < 'm'][c(TRUE, FALSE, FALSE)]
S
Sacha Epskamp

我认为您在问两件不一定相同的事情

我想提取原始的每 6 个元素

您可以通过索引序列来做到这一点:

foo <- 1:120
foo[1:20*6]

我想创建一个向量,其中每个元素都是另一个向量的第 i+6 个元素。

一个简单的方法是用 FALSE 补充一个逻辑因子,直到 i+6

foo <- 1:120
i <- 1
foo[1:(i+6)==(i+6)]
[1]   7  14  21  28  35  42  49  56  63  70  77  84  91  98 105 112 119

i <- 10
foo[1:(i+6)==(i+6)]
[1]  16  32  48  64  80  96 112

这么光滑!我也在另一个方向使用它,foo[1:(i+6)!=(i+6)] 即输出除第六个之外的所有值。
对向量 x 的每个第 n 个元素进行子集化的更一般的方法是 x[1:(length(x)/n)*n]
s
stevec

从向量中的任何起始位置选择每个第 n 个元素

nth_element <- function(vector, starting_position, n) { 
  vector[seq(starting_position, length(vector), n)] 
  }

# E.g.
vec <- 1:12

nth_element(vec, 1, 3)
# [1]  1  4  7 10

nth_element(vec, 2, 3)
# [1]  2  5  8 11

u
user11130854

要选择偏移量/偏移量为 f=0,...,n-1 的每个第 n 个元素,请使用

vec[mod(1:length(vec), n)==f]

当然,您可以将其包装在一个不错的函数中:

nth_element <- function(vec, interval, offset=0){
    vec[mod(1:length(vec), interval)==mod(offset, interval)]
}