在数据框中将列向上或向下移动 n 行

Question

假设我有以下 data.frame df:

set.seed(1)
df = data.frame(a = rnorm(10), 
                b = rnorm(10), 
                c = rnorm(10), 
                d = rnorm(10))

> df
            a           b           c           d
1  -0.6264538  1.51178117  0.91897737  1.35867955
2   0.1836433  0.38984324  0.78213630 -0.10278773
3  -0.8356286 -0.62124058  0.07456498  0.38767161
4   1.5952808 -2.21469989 -1.98935170 -0.05380504
5   0.3295078  1.12493092  0.61982575 -1.37705956
6  -0.8204684 -0.04493361 -0.05612874 -0.41499456
7   0.4874291 -0.01619026 -0.15579551 -0.39428995
8   0.7383247  0.94383621 -1.47075238 -0.05931340
9   0.5757814  0.82122120 -0.47815006  1.10002537
10 -0.3053884  0.59390132  0.41794156  0.76317575

我希望（仅）将 b、c 和 d 列向上移动 3 行。 a 列必须保持固定，删除 NAs 的行。

这是我想要的 3 行移位输出：

> df_shift_3
            a           b           c           d
1  -0.6264538 -2.21469989 -1.98935170 -0.05380504 
2   0.1836433  1.12493092  0.61982575 -1.37705956
3  -0.8356286 -0.04493361 -0.05612874 -0.41499456
4   1.5952808 -0.01619026 -0.15579551 -0.39428995
5   0.3295078  0.94383621 -1.47075238 -0.05931340
6  -0.8204684  0.82122120 -0.47815006  1.10002537
7   0.4874291  0.59390132  0.41794156  0.76317575

我如何创建一个函数，将此输出概括为 n 行的 up 或 down 移位？

我尝试使用以下功能https://rdrr.io/cran/useful/man/shift.column.html但没有成功。

感谢您的帮助

Answer 1

dplyr 的 lead/lag 函数对此很有帮助。 df %>% mutate_at(-1, ~lead(., 3)) %>% slice(1:(n()-3))

你可以把它变成一个函数

shift_all_but_first <- function(data, N) data %>% mutate_at(-1, ~lead(., N)) %>% slice(1:(n()-N))

shift_all_but_first(df, 3)

在数据框中将列向上或向下移动 n 行

shift columns up or down by n rows in dataframe

r

rows

shift

multiple-columns

dataframe