python - 将(设置)应用于 Pandas 数据框中的两列

Question

我在数据框中有三列。我想让第二列和第三列有集合而不是列表。使用 df['column_name_2 和 column_name_3'].apply(set) 会导致错误，因为据我所知，此函数只会接受 1 个参数。然而，将它们分成两个独立的功能，完全消除了第三列。

这是我的:

    column_1     column_2     column_3
       1         [lk, 18m]    [kjaf]

这就是我想要的:

    column_1     column_2     column_3
       1         {lk, 18m}    {kjaf}

Answer 1

我认为需要在嵌套列表中定义列，然后 applymap使用设置:

df[['column_2', 'column_3']] = df[['column_2', 'column_3']].applymap(set)

或者使用循环:

cols = ['column_2', 'column_3']
for c in cols:
    df[c] = df[c].apply(set)

示例:

df = pd.DataFrame({'column_1': [1, 1], 
                   'column_2': [['lk', '18m'], ['lk', 'r']],
                   'column_3': [['kjaf'], ['ddd']]})

print (df)
   column_1   column_2 column_3
0         1  [lk, 18m]   [kjaf]
1         1    [lk, r]    [ddd]

df[['column_2', 'column_3']] = df[['column_2', 'column_3']].applymap(set)
print (df)
   column_1   column_2 column_3
0         1  {18m, lk}   {kjaf}
1         1    {r, lk}    {ddd}