跨多个数据框列查找全部

Question

data = {'Cat':  ['A Phaser','A','B Phaser','B','B','B'],
        'L1': ['Phase','xyzss','xyzss','Phase','xyzss','xyzss'],
        'L2': ['xyzss','Stage','xyzss','xyzss','Phase2','xyzss'],
        'L3': ['xyzss','xyzss','xyzss','xyzss','xyzss','Step'],
        }

df = pd.DataFrame (data, columns = ['Cat','L1','L2','L3'])

def funt(s):
    if re.findall(r'Phase', s, re.IGNORECASE):
        return 'Phase'
    elif re.findall(r'Stag', s, re.IGNORECASE): 
        return 'Stage'
    elif re.findall(r'Step', s, re.IGNORECASE): 
        return 'Step'
    
df[['L1','L2','L3']].apply(lambda row: '_'.join(row.values.astype(str)), axis=1).apply(lambda x : funt(x))

输出：

0    Phase
1    Stage
2     None
3    Phase
4    Phase
5     Step
dtype: object

我想知道是否有另一种方法可以解决这个问题，例如跨列应用 findall 而无需将列连接在一起？提前致谢！

Answer 1

过滤所需的行。使用替换，使 xyzss 为空。堆叠并重置索引，您将得到作为 pd 的结果。系列.

选项 1：如果 xyzss 不变：df['filter']=df.iloc[:,1:4].replace({'xyzss':np.nan}).stack().reset_index(drop=True)

选项 1：如果 xyzss 变化：

df.join(pd.Series(df.mask(~df.isin(pat), np.nan).stack().reset_index(level=1, drop=True),name='filter'))

跨多个数据框列查找全部

Findall across multiple dataframe columns

python

findall

pandas