熊猫索引和列创建

Question

我有一个数据集df 。

我根据特定条件从df提取了另一个数据集df_rec 。

我可以通过df_rec.index.访问df_rec的索引df_rec.index.

现在，我想在df中创建一列，如果与df_rec中的索引匹配，则df中的索引应填充为1，否则填充为0。

任何帮助将不胜感激。

我在想，这会引发错误。

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

Answer 1

您可以使用df索引上的map来检查它是否在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]