如何使用R中的for循环将字符串列表转换为数据框名称？

Question

我希望对 R 中的这个问题有一些建议。我有一个数据框“my_fruits_data”，其中包含许多列，包括 name_cols 中的索引列。我想用 for 循环一一过滤这些索引列，并将过滤后的记录存储在各自的数据框中，它们的名称列在 df_fruits 中以进行后处理。显然，它不起作用，因为 df_fruits 元素是字符串而不是实际的数据框名称。我已经搜索并得到了一些提示，但实际上都没有帮助。

# column names
name_cols <- c("Index_apple",  
             "Index_pear",
             "Index_orange",  
             "Index_watermelon",
             "Index_strawberry"
         )
# dataframe names for filtered result 
df_fruits <- c("df_apple",  
             "df_pear",
             "df_orange",  
             "df_watermelon",
             "df_strawberry")

for (i in name_cols) 
{  
    df_fruits[i] <- my_fruits_data %>% 
           filter (.data[[name_cols[i]]] ==1) 
    ......
}

谢谢 追逐77

Answer 1

我了解到您希望根据水果类型拆分数据，该类型由单独的索引列提供。这是使用示例数据集执行此操作的方法。

library(tidyverse)
my_fruits_data = tribble(
  ~ index_apple, ~ index_pear, ~index_banana, ~ x1,
  1, 0, 0, 10,
  1, 0, 0, 11,
  0, 1, 0, 12,
  0, 0, 1, 13,
  0, 0, 1, 14, 
  0, 0, 1, 15
)

示例数据：

> my_fruits_data
# A tibble: 6 x 4
  index_apple index_pear index_banana    x1
        <dbl>      <dbl>        <dbl> <dbl>
1           1          0            0    10
2           1          0            0    11
3           0          1            0    12
4           0          0            1    13
5           0          0            1    14
6           0          0            1    15

首先，您可以将数据转换为有一个提到水果类型的 fruit 列：

fruit_data = my_fruits_data %>% 
  pivot_longer(
    cols = starts_with("index_"), 
    names_prefix = "index_", 
    names_to = "fruit",
    values_to = "fruit_ind"
  ) %>% 
  filter(fruit_ind == 1) %>% 
  select(-fruit_ind)

结果：

> fruit_data
# A tibble: 6 x 2
     x1 fruit 
  <dbl> <chr> 
1    10 apple 
2    11 apple 
3    12 pear  
4    13 banana
5    14 banana
6    15 banana

最后，正如@Onyambu 提到的，您可以考虑按我们的新变量fruit 对这些数据进行分组。如果你想对不同的水果做不同的处理，你可以split()数据来​​获取每个水果的单独数据框列表：

> split(fruit_data, fruit_data$fruit)
$apple
# A tibble: 2 x 2
     x1 fruit
  <dbl> <chr>
1    10 apple
2    11 apple

$banana
# A tibble: 3 x 2
     x1 fruit 
  <dbl> <chr> 
1    13 banana
2    14 banana
3    15 banana

$pear
# A tibble: 1 x 2
     x1 fruit
  <dbl> <chr>
1    12 pear