条件数据框选择的矢量化解决方案

Question

我最近问了一个问题 - How do I add conditionally to a selection of cells in a pandas dataframe column when the the column is a series of lists?，但我相信有一个我以前没有考虑过的新问题。

在以下数据框中，我需要两个条件来更改列 d。 d 列中的每个值都是list。

• 其中a == b，d 中的最后一个整数加一。

• 其中a != b，整数列表被扩展，值1被插入到d列中list的末尾。

  a       b       c           d           
  On      On      [0]         [0,3]       
  On      Off     [0]         [0,1]
  On      On      [0]         [2]         
  On      On      [0]         [0,4,4]         
  On      Off     [0]         [0]
  

• 因此，数据框会是这样的：

  a       b       c       d       
  On      On      [0]     [0,4]       
  On      Off     [0]     [0,1,1]     
  On      On      [0]     [3]
  On      On      [0]     [0,4,5] 
  On      Off     [0]     [0,1]
  

我意识到这可以使用 pd.Series.apply 方法结合预定义函数或使用 lambda 来完成，但是数据框由 100000 行组成，我希望这两个条件的矢量化解决方案可能存在。

Answer 1

作为 Edchum says，vecorised 解决方案可能存在问题。

一个带有apply自定义functions的非矢量化解决方案：

df['e'] = df['d']

def exten(lst):
    return lst + [1]

def incre(lst):
    lst[-1] = lst[-1] + 1
    return lst

df.loc[df.a != df.b, 'd'] = df.e.apply(exten)
df.loc[df.a == df.b, 'd'] = df.e.apply(incre)
df = df.drop('e', axis=1)
print df
    a    b    c          d
0  On   On  [0]     [0, 4]
1  On  Off  [0]  [0, 1, 1]
2  On   On  [0]        [3]
3  On   On  [0]  [0, 4, 5]
4  On  Off  [0]     [0, 1]