Создать столбец, добавив столбцы, используя условие - PullRequest
0 голосов
/ 14 марта 2019

Я хотел бы добавить новый столбец в существующий фрейм данных, в котором я покажу сумму значений в других столбцах, за исключением того, что я не хочу включать одно из имен столбцов, которые я назначил в качестве переменной.

Например,

ignore = 'Column3'

Column1 Column2 Column3
1        4       2
2        10      2
3        15      1

Я хочу создать новый столбец и добавить только Column1 со столбцом, однако, как вы можете видеть, у меня есть ignore = 'Column3' в качестве переменной, есть ли способ добавить столбцы в код, но включить переменную ignore чтобы какой-либо столбец, использованный в переменной ignore, не был добавлен в расчет?

1 Ответ

3 голосов
/ 14 марта 2019

IIUC, вы можете использовать:

ignore='Column3'
df['New']=df.loc[:,df.columns!=ignore].sum(axis=1)
print(df)

   Column1  Column2  Column3  New
0        1        4        2    5
1        2       10        2   12
2        3       15        1   18

Или, если вы хотите, чтобы список столбцов игнорировался:

ignore=['Column3']
df['New']=df.loc[:,df.columns.difference(ignore)].sum(axis=1)
print(df)

   Column1  Column2  Column3  New
0        1        4        2    5
1        2       10        2   12
2        3       15        1   18
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...