то же самое, что и python pandas: как найти строки в одном фрейме данных, но не в другом? но с несколькими столбцами
Это настройка:
import pandas as pd
df = pd.DataFrame(dict(
col1=[0,1,1,2],
col2=['a','b','c','b'],
extra_col=['this','is','just','something']
))
other = pd.DataFrame(dict(
col1=[1,2],
col2=['b','c']
))
Теперь я хочу выбрать строки из df
, которые не существуют в других. Я хочу сделать выбор col1
и col2
В SQL я бы сделал:
select * from df
where not exists (
select * from other o
where df.col1 = o.col1 and
df.col2 = o.col2
)
И в Pandas я могу сделать что-то подобное, но он чувствует себя очень уродливо. Часть уродства можно было бы избежать, если бы у df был id-column, но он не всегда был доступен.
key_col = ['col1','col2']
df_with_idx = df.reset_index()
common = pd.merge(df_with_idx,other,on=key_col)['index']
mask = df_with_idx['index'].isin(common)
desired_result = df_with_idx[~mask].drop('index',axis=1)
Итак, может быть, есть более элегантный способ?