有没有办法根据列标签中的文本有条件地添加多个pandas列？

import pandas as pd

# 创建一个示例DataFrame
data = {'A_文本列': ['foo', 'bar', 'baz'],
        'B_数字列': [1, 2, 3]}
df = pd.DataFrame(data)

# 定义要添加的列的条件
text_to_find = '文本'

# 使用条件语句创建布尔索引
mask = df.columns.str.contains(text_to_find)

# 使用apply函数添加新的列
df[['新列1', '新列2']] = df[mask].apply(lambda row: [row['A_文本列'] + '_新列1', row['B_数字列'] * 2], axis=1)

# 打印结果
print(df)

  A_文本列  B_数字列   新列1  新列2
0    foo      1  foo_新列1    2
1    bar      2  bar_新列1    4
2    baz      3  baz_新列1    6