python - 比较数据框中的两列python

Question

我正在尝试比较数据框中的一行是否为某个值。

例如: if word = 'bleu' and ink = 'blue' ，我想在我的数据框中添加一个新列 'congruent' = 1和if not = 0 .

我写了这段代码:

import pandas as pd
import numpy as np
import random

word = ['bleu', 'rouge', 'vert', 'jaune']

ink = ['blue', 'red', 'green', 'yellow']

my_list_word = [random.choice(word) for _ in range(60)]

df = pd.DataFrame(my_list_word, columns=["word"])

my_list_ink = [random.choice(ink) for _ in range(60)]

df['ink'] = my_list_ink

df['congruent'] = 0

print(df)

我尝试做一个循环，但没有成功，我的全等列全为 0。

有人知道怎么做吗？

谢谢!

Answer 1

我认为您正在寻找映射和匹配，即

di = dict(zip(word,ink))
df['new'] = (df['word'].map(di) == df['ink']).astype(int)

    word     ink  new
0   bleu     red    0
1  jaune  yellow    1
2  jaune    blue    0
3  jaune  yellow    1
4   vert    blue    0