python - 根据条件拆分列值

Question

如何根据 Pandas 中的条件拆分列的值？
如果 A 中的值小于 5，我想将 B 列转换为 point(.) 之前的第一个拆分
例如，我有下表:


一种
乙


2
ABC.DEF

2
ABC.DEF

8
ABC.DEF

8
ABC.DEF


它应该更改为


一种
乙


2
美国广播公司

2
美国广播公司

8
ABC.DEF

8
ABC.DEF

Answer 1

尝试:

m=df['A'].lt(5)
#your condition

最后:

df.loc[m,'B']=df.loc[m,'B'].str.split('.').str[0]

或者

#import numpy as np
df['B']=np.where(m,df['B'].str.split('.').str[0],df['B'])

df 的输出:

    A   B
0   2   ABC
1   2   ABC
2   8   ABC.DEF
3   8   ABC.DEF