Панды - добавить столбец с суммой значений четных строк - PullRequest
0 голосов
/ 28 ноября 2018

У меня есть фрейм данных, и я хочу новый столбец с суммой других столбцов в той же строке.Критерии: Я хочу Нэн, если сумма нечетная, а сумма, если значение четное.

df = pd.DataFrame([[1, 8, 7, 2],
                  [8, 5, 9, 4],
                  [1, -5, 3, -2]], columns=list('pqrs'))
df

    p   q   r   s
0   1   8   7   2
1   8   5   9   4
2   1   -5  3   -2

Expected output:
    p   q   r   s   t
0   1   8   7   2   18.0
1   8   5   9   4   26.0
2   1   -5  3   -2  NaN

1 Ответ

0 голосов
/ 28 ноября 2018

Использование np.where

df['new']=np.where(df.sum(1)%2==0,df.sum(1),np.nan)
df
   p  q  r  s   new
0  1  8  7  2  18.0
1  8  5  9  4  26.0
2  1 -5  3 -2   NaN
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...