如何使用pandas合并CSV两列并生成逗号分隔的列表格式结果

你可以通过以下代码实现需求，根据你现有列的存储格式选择对应方案即可：

场景1：两列均为原生列表类型 #

如果你的bboxes和class_names列本身存储的就是Python列表对象，直接执行：

import pandas as pd

df['final_bboxes'] = df.apply(lambda row: [row['bboxes'], row['class_names']], axis=1)

场景2：两列为字符串格式的列表 #

如果是从csv等文件读取后，列内容为看起来像列表的字符串，需要先转换为真实列表再合并：

import ast
import pandas as pd

# 转换字符串为列表
df['bboxes'] = df['bboxes'].apply(ast.literal_eval)
df['class_names'] = df['class_names'].apply(ast.literal_eval)

# 合并生成新列
df['final_bboxes'] = df.apply(lambda row: [row['bboxes'], row['class_names']], axis=1)

以上代码运行后生成的final_bboxes列，完全符合要求的格式：两个原列值用逗号分隔，整体用方括号包裹。

内容的提问来源于stack exchange，提问作者Engr Ali