給定存盤行和列的兩個向量，如何選擇資料框的值？為了mtcars
mpg cyl disp hp drat wt qsec vs am gear carb
Mazda RX4 21.0 6 160.0 110 3.90 2.620 16.46 0 1 4 4
Mazda RX4 Wag 21.0 6 160.0 110 3.90 2.875 17.02 0 1 4 4
Datsun 710 22.8 4 108.0 93 3.85 2.320 18.61 1 1 4 1
Hornet 4 Drive 21.4 6 258.0 110 3.08 3.215 19.44 1 0 3 1
Hornet Sportabout 18.7 8 360.0 175 3.15 3.440 17.02 0 0 3 2
Valiant 18.1 6 225.0 105 2.76 3.460 20.22 1 0 3 1
Duster 360 14.3 8 360.0 245 3.21 3.570 15.84 0 0 3 4
Merc 240D 24.4 4 146.7 62 3.69 3.190 20.00 1 0 4 2
Merc 230 22.8 4 140.8 95 3.92 3.150 22.90 1 0 4 2
...
我試過了
data.frame(row = c(1, 3, 5),
col = c('mpg', 'cyl', 'disp')) |>
mutate(value = mtcars[row, col])
并期望得到
row col value
1 1 mpg 21.0
2 3 cyl 4
3 5 disp 360.0
代替
row col value.mpg value.cyl value.disp
1 1 mpg 21.0 6 160
2 3 cyl 22.8 4 108
3 5 disp 18.7 8 360
uj5u.com熱心網友回復：
矩陣索引來救援。見?Extract：
第三種索引形式是通過一個數字矩陣，每個維度有一列：索引矩陣的每一行然后選擇陣列的單個元素，結果是一個向量。
...
如果陣列具有維度名稱，則還支持通過每個維度一列的字符矩陣進行索引。
所以：
df <- data.frame(row = c(1, 3, 5),
col = c('mpg', 'cyl', 'disp'))
## character matrix indexing
cbind(df, value = mtcars[cbind(rownames(mtcars)[df$row], df$col)])
## numeric matrix indexing
cbind(df, value = mtcars[cbind(df$row, match(df$col, colnames(mtcars)))])
## both resulting in:
## row col value
##1 1 mpg 21
##2 3 cyl 4
##3 5 disp 360
正如@Onyambu 所說，transform如果您愿意，還可以使用 /base R 管道：
df |> transform(value = mtcars[cbind(rownames(mtcars)[row], col)])
...或dplyr管道mutate：
df %>% mutate(value = mtcars[cbind(rownames(mtcars)[row], col)])
uj5u.com熱心網友回復：
在基礎 R 中應用族
一種方法是使用sapply查詢資料框中的所有行，并使用每行的row和資訊來索引資料集。colmtcars
編輯：感謝@Darren Tsai 提供的mapply解決方案。
library(dplyr)
data.frame(row = c(1, 3, 5),
col = c('mpg', 'cyl', 'disp')) %>%
mutate(value = sapply(1:nrow(.), \(x) mtcars[.[x, 1], .[x, 2]]))
# or mapply
mutate(value = mapply(\(x, y) mtcars[x, y], row, col))
來自 dplyr 的 rowwise()
另一種方法是使用rowwise()：
library(dplyr)
data.frame(row = c(1, 3, 5),
col = c('mpg', 'cyl', 'disp')) %>%
rowwise() %>%
mutate(value = mtcars[row, col])
輸出
row col value
1 1 mpg 21
2 3 cyl 4
3 5 disp 360
uj5u.com熱心網友回復：
這是另一種tidyverse選擇：
library(tidyverse)
data.frame(row = c(1, 3, 5),
col = c('mpg', 'cyl', 'disp')) %>%
mutate(value = map2_dbl(row, col, ~ mtcars[.x, .y]))
輸出
row col value
1 1 mpg 21
2 3 cyl 4
3 5 disp 360
uj5u.com熱心網友回復：
另一種方法是用來diag獲取值
rows <- c(1, 3, 5)
cols <- c("mpg", "cyl", "disp")
data.frame(
row = rows,
col = cols,
value = diag(as.matrix(mtcars[rows, cols]))
)
row col value
1 1 mpg 21
2 3 cyl 4
3 5 disp 360
轉載請註明出處，本文鏈接：https://www.uj5u.com/qianduan/491322.html