为每行创建一个列，其中列表值来自以数据帧中的子字符串开头的列

import pandas as pd

# 假设我们有以下的DataFrame
data = {
    'A': ['foo_1', 'bar_2', 'baz_3'],
    'B': ['foo_4', 'bar_5', 'baz_6'],
    'C': ['foo_7', 'bar_8', 'baz_9']
}
df = pd.DataFrame(data)

# 创建一个空的字典来存储结果
result = {}

# 遍历DataFrame的列
for column in df.columns:
    # 使用str.startswith方法检查每个元素是否以'foo_'开头
    mask = df[column].str.startswith('foo_')
    # 如果是，将对应的值添加到结果字典中
    if mask.any():
        result[column] = df.loc[mask, column].tolist()

# 将结果字典转换为DataFrame
result_df = pd.DataFrame(result)

print(result_df)

     A    B
0  [foo_1]  [foo_4]
1     NaN  [foo_7]