Добавление данных с толбец, через цикл

Question

Имеется 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 не понимаю.

Answer 1

Если я правильно понял вопрос:

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

БЛОГ НА HUSL

Добавление данных с толбец, через цикл

Ответы (1 шт):