主页 / 问题 / 1305799
Accepted
Nevskiy
Asked:2022-07-16 15:51:22 +0000 UTC2022-07-16 15:51:22 +0000 UTC

通过循环从列中添加数据[关闭]

  • 772

有3列。您需要比较 COL1 和 COL2，找到相同的结果。比较后，将结果写入COL3。

该文件是原始文件。

在此处输入图像描述

因此，该文件应如下所示。

在此处输入图像描述

import pandas as pd

test1 = pd.read_excel('test1.xlsx', sheet_name='Sheet1', usecols=['COL1', 'COL2', 'COL3'])

for i in test1['COL1']:
    for j in test1['COL2']:
        if(i == j):
            print(j)

从逻辑上讲，数据比较正确，数字 1-11 显示在输出中。但是我不明白怎么把结果写在COL3的第三列。

python

1 个回答

  1. Best Answer

    如果我正确理解了这个问题：

    In [1]: df = pd.DataFrame({"a":[1,2,3], "b":[2,3,4]})

In [2]: df
Out[2]:
   a  b
0  1  2
1  2  3
2  3  4

In [3]: np.intersect1d(df["a"], df["b"])
Out[3]: array([2, 3])

In [4]: isect = np.intersect1d(df["a"], df["b"])

In [5]: df.loc[df["a"].isin(isect), "res"] = df["a"]

In [6]: df
Out[6]:
   a  b  res
0  1  2  NaN
1  2  3  2.0
2  3  4  3.0
    • 2

相关问题