在 Pandas 数据框中创建一个新的列表列，其中包含来自另一列的唯一值

Question

我有一个包含一列列表的数据框：

    full_list_to_check
 0          NaN 
 1          NaN 
 2    [1, 2, 3, 4, 5] 
 3        [6, 6] 
 4        [11, 11] 

我需要创建一个新列，如果列表中存在重复项，它会为每一行显示一个不同的列表，否则只是相同的列表。

  full_list_to_check            new_col
 0          NaN                   NaN
 1          NaN                   NaN
 2    [1, 2, 3, 4, 5]           [1, 2, 3, 4, 5]
 3        [6, 6]                  [6]
 4        [11, 11]                [11]

我试过这个：

df['new_col'] = df['full_list_to_check'].apply(lambda x: list(set(x)))

但我收到此错误：

TypeError: 'float' object is not iterable

Answer 1

你必须检查Nan ：

df['full_list_to_check'].apply(lambda x: list(set(x)) if not np.any(pd.isna(x)) else np.nan)

更新：

df['full_list_to_check'].apply(lambda x: list(set(x)) if x is not np.nan else np.nan)

0                NaN
1                NaN
2    [1, 2, 3, 4, 5]
3                [6]
4               [11]

Answer 2

你可以试试这个：

df['new_col'] = df.loc[~df['full_list_to_check'].isna(), 'full_list_to_check'].apply(lambda x: list(set(x)))

full_list_to_check new_col
0 NaN              NaN
1 NaN              NaN
2 [1, 2, 3, 4, 5]  [1, 2, 3, 4, 5]
3 [6, 6]           [6]
4 [11, 11]         [11]

Answer 3

你可以使用：

df['new_col'] = df['full_list_to_check'].apply(lambda x: list(set(x)) if isinstance(x,list) else x)

其他答案仅在您的数据中没有其他值然后列出或 NaN 时才有效。