如何根據python中其他列的條件將字符串列拆分為另一個列？

Question

我有以下數據框：

import pandas as pd
 dt = pd.DataFrame({'col': ['A','A_B']})

我希望col==A的rows值all放入一個新列（ col2 ）中，其余的進行str.split操作

我希望最終的 df 看起來像這樣：

  dt = pd.DataFrame({'col': ['A', 'A_B'],
                         'col2': ['all', 'B']})

我試過：

dt['col2'] = np.where(dt.col == 'A', 'all',
                                 dt.col.apply(lambda x: x.split('_')[1]))

但我收到此錯誤： IndexError: list index out of range

Answer 1

如果使用 pandas 文本函數，它在您的情況下工作正常 - 返回缺失值，因為第二個列表不存在：

print (dt.col.str.split('_').str[1])
0    NaN
1      B
Name: col, dtype: object


dt['col2'] = np.where(dt.col == 'A', 'all', dt.col.str.split('_').str[1])
print (dt)
   col col2
0    A  all
1  A_B    B

或使用[-1]選擇拆分后的最后一個列表：

dt['col2'] = np.where(dt.col == 'A',  'all',
                      dt.col.apply(lambda x: x.split('_')[-1]))

或者是通過反轉掩碼可能的過濾器值：

m = dt.col == 'A'
dt['col2'] = np.where(m, 'all',
                      dt.loc[~m, 'col'].apply(lambda x: x.split('_')[1]))

Answer 2

你可以

dt['col2'] = dt.col.str.split('_', expand = True).fillna('all')[1]