如何使用Python pandas将DataFrame列中的列表元素转换为元组

问题场景 #

现有存储列表类型元素的DataFrame，原始结构如下：

|  B       |  A         |
  +----------+------------+
  |0.07      | [a,b,c]    |
  |0.009     | [d,e,f]    |
  |0.008     | [g,h,j]    |
  +----------+------------+

需要将A列中存储的所有列表元素转换为元组类型，预期输出结构如下：

|  B       |  A         |
  +----------+------------+
  |0.07      | (a,b,c)    |
  |0.009     | (d,e,f)    |
  |0.008     | (g,h,j)    |
  +----------+------------+

实现代码 #

直接对目标列使用apply方法，映射Python内置的tuple构造函数即可逐行完成转换，代码如下：

import pandas as pd

# 构造示例数据（和样例结构一致）
df = pd.DataFrame({
    'B': [0.07, 0.009, 0.008],
    'A': [['a', 'b', 'c'], ['d', 'e', 'f'], ['g', 'h', 'j']]
})

# 核心转换逻辑：逐行将A列的列表转为元组
df['A'] = df['A'].apply(tuple)

执行上述代码后打印df，即可得到预期的输出结果，可以通过如下代码验证转换后的元素类型：

# 验证第一行A列元素的类型
print(type(df.loc[0, 'A']))
# 输出：<class 'tuple'>

避坑提示 #

• 不要直接使用df['A'].astype(tuple)做转换：该方法会将整列所有数据合并为一个大元组，无法实现逐行转换单个列表的效果。
• 如果需要同时转换多列的列表为元组，可以对多列组成的子DataFrame使用applymap(tuple)，例如df[['A', '其他列名']] = df[['A', '其他列名']].applymap(tuple)。

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