Добавить столбец в фрейм данных на основе других столбцов - PullRequest
0 голосов
/ 19 июня 2020

Я хотел бы скопировать содержимое col_B в новый столбец только тогда, когда содержимое col_A равно numeri c, а не NaN. Я понимаю, что это может показаться рудиментарным, но я новичок в python. Заранее спасибо

  col_A  col_B  col_C  col_D

0      NaN    2      a      b
1      2      4      b      d
2      NaN    1      c      a
3      4      2      d      b
4      5      3      e      c

1 Ответ

1 голос
/ 19 июня 2020

Вы можете использовать numpy. Где здесь:

df['new_col'] = np.where(~df['col_A'].isna(), df['col_B'], '')
print(df)

   col_A  col_B col_C col_D new_col
0    NaN      2     a     b
1    2.0      4     b     d       4
2    NaN      1     c     a
3    4.0      2     d     b       2
4    5.0      3     e     c       3
...