Python中从Pandas DataFrame读取元组列的问题

问题原因 #

CSV文件中的RGB值以带引号的字符串格式存储（如"(0, 0, 0)"），Pandas默认将其识别为字符串类型，而非元组，导致无法直接通过索引获取颜色分量。

解决方案 #

方法1：用ast.literal_eval转换为元组

利用ast.literal_eval可将符合Python字面量格式的字符串转换为对应数据结构（如元组），保留RGB的元组形式：

import pandas as pd
import ast

# 读取CSV文件
df = pd.read_csv('Pinturas GW.csv')
# 将RGB列的字符串转换为元组
df['RGB Value'] = df['RGB Value'].apply(ast.literal_eval)

# 提取第一个颜料的红色分量
print(df['RGB Value'][0][0])  # 输出：0

方法2：拆分字符串生成独立的RGB分量列

如果需要单独处理红、绿、蓝三个分量，直接拆分字符串生成新列更便于后续操作：

import pandas as pd

df = pd.read_csv('Pinturas GW.csv')
# 移除括号并按逗号拆分，转换为整数类型
rgb_parts = df['RGB Value'].str.strip('()').str.split(', ', expand=True).astype(int)
# 给新列命名
rgb_parts.columns = ['Red', 'Green', 'Blue']
# 将分量列合并到原DataFrame
df = pd.concat([df, rgb_parts], axis=1)

# 提取第一个颜料的红色分量
print(df['Red'][0])  # 输出：0

方法3：读取时直接指定转换器

在read_csv阶段通过converters参数完成转换，一步到位：

import pandas as pd
import ast

# 读取时直接转换RGB列
df = pd.read_csv('Pinturas GW.csv', converters={'RGB Value': ast.literal_eval})

# 提取第一个颜料的红色分量
print(df['RGB Value'][0][0])  # 输出：0

内容的提问来源于stack exchange，提问作者Juan Ignacio Ochotorena