查找列值在其他列中匹配的索引

Question

我有一个包含两个整数列的数据框，例如：

   a  b
0  5  7
1  3  5
2  7  1

我需要一个包含索引的附加列，其中列a 的值等于当前行的列b 的值。即：b=7 与索引2 处的a=7 匹配，b=5 与索引a=5 处的a=5 匹配，b=1 不匹配。期望的输出：

   a  b  c
0  5  7  2
1  3  5  0
2  7  1  NaN

永远不会有多个行满足条件。

Answer 1

searchsorted 选项：

# sort column a and find candidate position of values in b in a
df.sort_values('a', inplace=True)
pos = df.a.searchsorted(df.b)

# handle edge case when the pos is out of bound
pos[pos == len(pos)] = len(pos) - 1

# assign the index to column as c and mark values that don't match as nan
df['c'] = df.index[pos]
df.loc[df.a.loc[df.c].values != df.b.values, 'c'] = np.nan

df.sort_index()

#   a  b    c
#0  5  7  2.0
#1  3  5  0.0
#2  7  1  NaN