熊猫索引和列创建

Question

我有一个数据集df。

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

我可以通过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

您可以使用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]