Индексация панд и создание столбцов

Question

У меня есть набор данных, df.

Я извлек другой набор данных из df, df_rec, основываясь на определенном условии.

Я могу получить доступ к индексам df_rec по df_rec.index.

Теперь я хочу создать столбец в df, где индекс в df, если совпадает с индексами в df_rec, должен быть заполнен как 1, иначе 0.

Любая помощь, будет оценена.

Я думаю, что-то вроде ошибки.

df['reccurences'] = 0
df['reccurences'][df.index in df_rec.index] = 1

Answer 1

Вы можете использовать map для индекса df, чтобы проверить, находится ли он в df_res, и установить соответствующее значение, как показано ниже.

df = pd.DataFrame()

df['X'] = [1, 2, 3, 4, 5, 6]
df['Y'] = [10, 20, 30, 40, 50, 60]

df_res = df.loc[df['X'] > 3]

df['C'] = df.index.map(lambda x : 1 if x in df_res.index else 0)

ИЛИ вы можете сделать это

df['C'] = [1 if x in df_res.index else 0 for x in df.index]