python - pandas dataframe 基于另一个包含名称列的文件创建一个新的二进制事件列

Question

                  LigandName    Score
0   _017_model1/_017_model1     -10.54
1   _010_model1/_010_model1     -9.49
2   _113_model1/_113_model1     -9.40
3   _009_model1/_009_model1     -9.37
4   _193_model1/_193_model1     -9.36

这是数据帧的前几行。我想添加一个名为 IsActive 的新列，它基于该文件中是否存在配体名称。这是文件

0   _017_model1/_017_model1
1   _010_model1/_010_model1
2   _113_model1/_113_model1
3   _009_model1/_009_model1
4   _004_model1/_004_model1

如果配体名称存在，则 IsActive 条目应为 1，如果不存在，则应为 0。我该怎么办？

Answer 1

通过 Series.isin 创建 bool 掩码并转换为整数 - True 为 1，False 为 0:

print (df1)
                LigandName
0  _017_model1/_017_model1
1  _010_model1/_010_model1
2  _113_model1/_113_model1
3  _009_model1/_009_model1
4  _004_model1/_004_model1

df['IsActive'] = df['LigandName'].isin(df1['LigandName']).astype(int)
print (df)
                LigandName  Score  IsActive
0  _017_model1/_017_model1 -10.54         1
1  _010_model1/_010_model1  -9.49         1
2  _113_model1/_113_model1  -9.40         1
3  _009_model1/_009_model1  -9.37         1
4  _193_model1/_193_model1  -9.36         0