Question

我有两个带有客户ID的数据框（标有＆＃34; C_ID＆＃34;）以及一年的访问次数。

如果客户也在2009年购物，我想在我的2010数据框中添加一个列。所以我需要创建一个循环来检查2010年的C_ID是否存在于2009年，添加1，否则为0.

我使用了这段代码并且没有工作:(没有错误信息，没有任何反应）

for row in df_2010.iterrows():
    #check if C_ID exists in the other dataframe
    check = df_2009[(df_2009['C_ID'] == row['C_ID'])]

    if check.empty:
        #ID not exist in 2009 file, add 0 in new column
        row['shopped2009'] = 0

    else:
        #ID exists in 2009 file, add 1 into same column
        row['shopped2009'] = 1

Answer 1

您可以使用dataframe.isin（）

% timeit df_2010['new'] = np.where(df_2010['C_ID'].isin(df_2009['C_ID']), 1, 0)

最佳3：每循环384μs

正如@Kris建议的那样

%timeit df_2010['new'] = (df_2010['C_ID'].isin(df_2009['C_ID'])).astype(int)

最佳3：每循环584μs

或者

df_2010['new'] = df_2010['C_ID'].isin(df_2009['C_ID'])

Python Pandas数据帧添加＆＃34; 1＆＃34;如果ID存在于其他数据框中，则在新列中

1 个答案: