基于同一索引上的另一列值按系列拆分字符串

Question

你好，我有熊猫数据框

    code                                 LocationCod
0.  999010001-R-XQK33-4658-999-          R-XQK33
1.  602444201-N-FKD2-000257999-999-      N-FKD2
2.  602016601-N-TAS-777-999-             N-TAS
3.  602010001-345-999-999-               345

我想根据column LocationCod中的值拆分column Code中的值，该值应该扩展到新的列CategCode和MainCode。required表如下所示

    code                                 LocationCod     CategCode           MainCode
0.  999010001-R-XQK33-4658-999-          R-XQK33         4658-999-           999010001
1.  602444201-N-FKD2-000257999-999-      N-FKD2          000257999-999-      602444201
2.  602016601-N-TAS-777-999-             N-TAS           777-999-            602016601
3.  602010001-345-999-999-               345             999-999-            602010001

请帮忙

Answer 1

试试这个：

df['CategCode']=[df.code.values[i].split(df.LocationCod.values[i])[-1][1:] for i in range(len(df))]
df['MainCode']=[df.code.values[i].split(df.LocationCod.values[i])[0][:-1] for i in range(len(df))]