如何将pandas dataframe中列表值的列转换为对应元素的同列格式

适用前提 #

本方案适用于每个列表单元格仅包含1个元素的场景，如你给出的示例结构：

原始列值：['Bb1']、[54]、[84.0]
转换后值：Bb1、54、84.0

实现方法 #

方法1：快速批量提取（推荐） #

使用pandas的str访问器直接提取列表第一个元素，代码简洁性能高：

# 手动指定要处理的列
process_cols = ['B', 'C', 'D']
df[process_cols] = df[process_cols].apply(lambda x: x.str[0])

如果需要自动识别所有列表类型的列，不需要手动指定：

for col in df.columns:
    if isinstance(df[col].iloc[0], list):
        df[col] = df[col].str[0]

方法2：带容错逻辑的实现 #

如果存在空列表、非列表值等异常情况，可加判断避免报错：

import pandas as pd

def extract_list_val(x):
    if isinstance(x, list) and len(x) > 0:
        return x[0]
    # 异常情况返回空值，可按需修改返回值
    return pd.NA

df[['B','C','D']] = df[['B','C','D']].applymap(extract_list_val)

特殊场景处理 #

如果你的列值是字符串格式的伪列表（比如从csv读入后看起来是"['Bb1']"，实际是字符串类型），需要先转成真实列表再处理：

import ast
# 把字符串形式的列表转为真实列表
df['B'] = df['B'].apply(ast.literal_eval)
# 再执行提取逻辑
df['B'] = df['B'].str[0]

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