如何在数据框中创建从列值的字符串列表

Question

我有此数据框架：

Text     feat1   feat2   feat3    feat4
string1    1       1       0        0
string2    0       0       0        1
string3    0       0       0        0

我想以这种方式创建其他两个列：

Text     feat1   feat2   feat3    feat4     all_feat            count_feat
string1    1       1       0        0       ["feat1","feat2"]       2
string2    0       0       0        1       ["feat4"]               1
string3    0       0       0        0       []                      0

在朱莉娅（Julia）做什么的最佳方法是什么？

Answer 1

这是这样做的可能方法之一：

julia> df
3×5 DataFrame
 Row │ Text     feat1  feat2  feat3  feat4
     │ String   Int64  Int64  Int64  Int64
─────┼─────────────────────────────────────
   1 │ string1      1      1      0      0
   2 │ string2      0      0      0      1
   3 │ string3      0      0      0      0

julia> transform(df,
                 AsTable(r"feat") =>
                 ByRow(x -> [string(k) for (k,v) in pairs(x) if v == 1]) =>
                 :all_feat,
                 r"feat" => (+) => :count_feat)
3×7 DataFrame
 Row │ Text     feat1  feat2  feat3  feat4  all_feat            count_feat
     │ String   Int64  Int64  Int64  Int64  Array…              Int64
─────┼─────────────────────────────────────────────────────────────────────
   1 │ string1      1      1      0      0  ["feat1", "feat2"]           2
   2 │ string2      0      0      0      1  ["feat4"]                    1
   3 │ string3      0      0      0      0  String[]                     0