读取带有列表的pandas中的列以创建新的分类列

import pandas as pd

# 创建包含列表的示例数据框
data = {'ID': [1, 2, 3],
        'Fruit': [['Apple', 'Banana'], ['Orange', 'Grape'], ['Watermelon', 'Pineapple']]}
df = pd.DataFrame(data)

# 定义函数来处理列表列并创建新的分类列
def process_list(row):
    # 将列表中的每个元素转换为字符串类型
    elements = [str(element) for element in row['Fruit']]
    # 返回一个逗号分隔的字符串作为新的分类列的值
    return ', '.join(elements)

# 使用apply函数调用处理函数，并使用assign函数创建新的分类列
df = df.assign(Category=df.apply(process_list, axis=1))

# 打印结果
print(df)

   ID                  Fruit                 Category
0   1        [Apple, Banana]            Apple, Banana
1   2       [Orange, Grape]           Orange, Grape
2   3  [Watermelon, Pineapple]  Watermelon, Pineapple