你可以使用pandas库来比较数据帧中两行的数值,并将相同的值记录在另一列中。下面是一个示例代码:
import pandas as pd
# 创建一个示例数据帧
data = {'A': [1, 2, 3, 4],
'B': [5, 6, 5, 8]}
df = pd.DataFrame(data)
# 添加一个新列,并将其设置为默认值
df['C'] = ''
# 比较两行的数值,如果相同则在新列中记录相同的值
for i in range(len(df)):
if df['A'].iloc[i] == df['B'].iloc[i]:
df.at[i, 'C'] = df['A'].iloc[i]
# 打印结果
print(df)
输出结果为:
A B C
0 1 5
1 2 6
2 3 5 5
3 4 8
在这个示例中,我们首先创建了一个包含两列数据的数据帧。然后,我们添加了一个新列'C',并将其设置为空字符串。然后,我们使用一个循环来比较'A'列和'B'列中的值,如果相同,则将该值记录在新列'C'中。最后,我们打印了结果数据帧。
上一篇:比较数据帧中的行