如何删除python数据帧中最后一个下划线后的字符串？

Question

我想从数据框中删除最后一个下划线之后的所有字符串。如果我在dataframe中查看我的数据

AA_XX,
AAA_BB_XX,
AA_BB_XYX,
AA_A_B_YXX

我想得到这个结果

AA,
AAA_BB,
AA_BB,
AA_A_B

Answer 1

这是另一种方法

import pandas as pd
data = {'s': ['AA_XX', 'AAA_BB_XX', 'AA_BB_XYX', 'AA_A_B_YXX']}
df = pd.DataFrame(data)

def cond1(s):
    temp_s = s.split('_')
    temp_len = len(temp_s)
    if len(temp_s) == 1:
        return temp_s
    else:
        return temp_s[:len(temp_s)-1]
df['result'] = df['s'].apply(cond1)