pandas DataFrame移除列内多余组合列表项仅保留单个氧化态组合

你需要对存储化合物氧化态组合的DataFrame做统一处理：无论Oxi_State列的每行列表中有多少个符合价态平衡的氧化态元组，仅保留第一个组合，移除其余多余项，最终输出格式和原格式保持一致（仍为列表包裹元组的形式）。

方法1：生成DataFrame后批量处理 #

这是改动最小的方案，在你现有代码的基础上，加1行处理逻辑即可：

import pandas as pd
import numpy as np

# 你原有的构造DataFrame代码
dataPd1 = pd.DataFrame(res)
# 加上dtype=object避免不规则列表触发numpy告警
data = np.array(solutions, dtype=object)
dataPd2 = pd.DataFrame(data = data)
df_os = pd.concat([dataPd1,dataPd2],axis=1)
df_os.columns = ["ABO3", "Oxi_State"]

# 新增处理逻辑：逐行取氧化态列表的第一个元素，包装为单元素列表匹配原格式
df_os["Oxi_State"] = df_os["Oxi_State"].apply(lambda x: [x[0]] if isinstance(x, list) and len(x)>=1 else x)

处理后输出完全符合预期：

ABO3    Oxi_State
0   CeTmO3  [(4, 2)]
1   CeLuO3  [(3, 3)]
2   CeRhO3  [(4, 2)]

对于你提到的存在4种氧化态组合的BrNO3行，处理后会自动保留第一个组合(7,-1)，最终值为[(7,-1)]，无需额外适配。

方法2：生成氧化态列表时提前过滤（更高效） #

如果你不需要保留所有可能的氧化态组合，可以在生成solutions列表时就只取第一个符合条件的组合，减少无效遍历，代码修改如下：

solutions = []
for x in range(len(final)):
    first_valid = None
    # 遍历氧化态找到第一个满足A+B=6的组合就终止循环
    for x1 in final[x][0]:
        for x2 in final[x][1]:
            if x1 + x2 == 6:
                first_valid = (x1, x2)
                break
        if first_valid:
            break
    solutions.append([first_valid])

修改后你后续构造DataFrame的代码不需要做任何调整，生成的df_os直接就是仅保留第一个氧化态组合的结果，在数据量大的时候运行效率更高。

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