如何按特定条件对DataFrame的逗号分隔列执行选择性explode？

针对需求，我们可以通过Pandas结合自定义函数实现目标，以下是具体步骤和代码：

实现思路 #

1. 对col1的字符串进行分割，同时去除元素中的多余空格；
2. 遍历分割后的元素，提取开头的所有a开头元素，遇到非a开头元素立即停止遍历；
3. 若提取结果为空，则填充默认值default；
4. 最后使用explode将列表展开为多行结构。

完整代码实现 #

import pandas as pd

# 构造示例DataFrame
df = pd.DataFrame({
    'ID': [1, 2, 3],
    'col1': ['a100,a101,b100,c100', 'a105,b100', 'b101, c104']
})

def extract_a_elements(s):
    # 分割字符串并去除每个元素的首尾空格
    elements = [elem.strip() for elem in s.split(',')]
    a_elements = []
    for elem in elements:
        if elem.startswith('a'):
            a_elements.append(elem)
        else:
            # 遇到非a开头元素，终止遍历
            break
    # 无a开头元素时返回默认值列表
    return a_elements if a_elements else ['default']

# 生成目标元素列
df['element'] = df['col1'].apply(extract_a_elements)
# 执行explode并保留需要的列
result = df.explode('element')[['ID', 'element']]

print(result)

执行结果 #

运行代码后输出结果与期望一致：

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