如何在DataFrame中创建一个在不同列中显示模式的新列？

import pandas as pd

# 创建一个示例DataFrame
df = pd.DataFrame({'col1': ['A', 'B', 'C', 'D'],
                   'col2': ['AAB', 'BBC', 'CDD', 'DDE'],
                   'col3': ['AAA', 'BBB', 'CCC', 'DDD']})

import re

# 定义一个函数来获取每个元素的模式
def get_pattern(element):
    pattern = re.compile(r'(\w)\1+')
    match = pattern.search(element)
    if match:
        return match.group()
    else:
        return 'No pattern'

# 在DataFrame中应用函数并创建新列
df['pattern'] = df.apply(lambda row: get_pattern(row['col2']), axis=1)

print(df)

  col1 col2 col3 pattern
0    A  AAB  AAA      AA
1    B  BBC  BBB      BB
2    C  CDD  CCC      CC
3    D  DDE  DDD      DD