如何提取pandas DataFrame中A列的唯一值列表？

可以通过以下步骤实现需求：

完整代码实现 #

import pandas as pd

data = {
    'A':['A1, B2, C', 'A2, A9, C', 'A3', 'A4, Z', 'A5, A1, Z'], 
    'B':['B1', 'B2', 'B3', 'B4', 'B4'], 
}
df = pd.DataFrame(data)

# 生成目标唯一值列表
list_A = df['A'].str.split(', ').explode().unique().tolist()
print(list_A)

输出结果 #

['A1', 'B2', 'C', 'A2', 'A9', 'A3', 'A4', 'Z', 'A5']

步骤说明 #

• df['A'].str.split(', ')：将A列每个字符串按, （逗号加空格）分割为子列表
• .explode()：把每个子列表的元素展开为单独行，得到包含所有元素的Series
• .unique()：从Series中筛选出所有唯一值
• .tolist()：将唯一值Series转换为列表

内容的提问来源于stack exchange，提问作者Alberto Alvarez