Вектовое решение для выбора условных данных

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 вставляется в конце list в колонке d.

  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] 
  

• В результате dataframe хотел бы это:

  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

As 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]