Сравнение левой колонки с правой колонкой в ​​Excel с питоном и пандами - PullRequest
0 голосов
/ 14 октября 2019

У меня есть вопрос о сравнении столбцов в Python. Например: если значение из столбца 1 начинается с «w», то добавьте «x» к той же строке в столбце B, но только если строка в B пуста. Я знаю, что должен быть цикл для фильтрации этих столбцов, но я совсем новичок с пандами.

    INPUT
    1   2
1   w   NaN
2   w   b
3   b   b
4   w   x

OUTPUT
    1   2
1   w   x
2   w   b
3   b   b
4   w   x

1 Ответ

0 голосов
/ 14 октября 2019

Сначала создайте оба условия с помощью isna() и str.startswith.

Затем используйте loc для заполнения столбца2:

m1 = df['1'].str.startswith('w')
m2 = df['2'].isna()

df.loc[m1&m2, '2'] = 'x'

Вывод

   1  2
1  w  x
2  w  b
3  b  b
4  w  x

note : я предположил, что ваши имена столбцов являются строками, поэтому яиспользуется '1' и '2'. Если они целые, используйте 1 и 2.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...