如何获取两列值匹配的位置? - Python

在数据处理的过程中，我们可能会遇到需要获取两列数值匹配的位置的需求。本文将介绍如何使用 Python 实现该功能。

首先，我们可以将两列数值转化为列表，并使用 enumerate 函数遍历其中的元素，寻找匹配的位置。

col1 = [1, 2, 3, 4, 5]
col2 = [3, 5, 7, 9, 8]

matches = []
for idx, val in enumerate(col1):
    if val in col2:
        matches.append((idx, col2.index(val)))

以上代码通过遍历 col1 列表，检查其中的元素是否存在于 col2 列表中，并记录下其在两列中的位置。matches 列表将存储位置信息（以元组的形式），其中第一个元素表示在 col1 中的位置，第二个元素表示在 col2 中的位置。

如果需要进一步处理匹配的位置信息，我们可以将其转化为字典或数据帧。下面是将匹配信息转化为字典的示例代码：

match_dict = {}
for pos in matches:
    match_dict[pos[0]] = pos[1]

对于较大的数据集，我们可以使用 Pandas 库中的数据帧来实现相同的功能，并获得更高的效率。

import pandas as pd

df = pd.DataFrame({'col1': [1, 2, 3, 4, 5], 'col2': [3, 5, 7, 9, 8]})

matches = df[df['col1'].isin(df['col2'])].reset_index(drop=True)

以上代码将 col1 和 col2 列转化为数据帧 df，并使用 isin 函数检查 col1 中是否存在于 col2 中的元素。最后，将选择出的匹配信息存储在数据帧 matches 中。

以上就是使用 Python 获取两列数值匹配的位置的实现方法。

参考资料：

• Python documentation: enumerate
• Pandas documentation: isin