1

我正在寻找一种pythonic方法来处理这个数据框列和列表匹配问题:

这是我的数据框:

在此处输入图像描述

现在,我有一个 roll_no 列表:

     roll_no_matching = [3,5]

我想在此列表和 B 列(即 roll_no)之间进行匹配。并插入一个新列,其中 0 表示不匹配,1 表示匹配。例如,生成的数据框将如下所示: 在此处输入图像描述

我知道如何根据任何索引列匹配两个数据框,并可能合并两个数据框中的一些列,但是我不确定如何创建一个新列。非常感谢这里的任何指导。

4

1 回答 1

0

用于通过Series.isin以下方式将掩码转换为整数的测试成员资格Series.astype

df['Merit_list'] = df['roll No.'].isin(roll_no_matching).astype(int)

或者Series.view

df['Merit_list'] = df['roll No.'].isin(roll_no_matching).view('i1')

或通过numpy.where

df['Merit_list'] = np.where(df['roll No.'].isin(roll_no_matching), 1, 0)
于 2021-01-07T09:24:22.167 回答