下面是一个示例代码,用于比较A、B和C列,并在匹配时返回D列的值:
import pandas as pd
# 创建示例数据
data = {'A': [1, 2, 3, 4, 5],
'B': ['a', 'b', 'c', 'd', 'e'],
'C': [True, False, True, True, False],
'D': ['apple', 'banana', 'orange', 'grape', 'watermelon']}
df = pd.DataFrame(data)
# 比较A、B和C列,并在匹配时返回D列的值
result = df.loc[(df['A'] == 3) & (df['B'] == 'c') & (df['C'] == True), 'D']
print(result)
输出结果为:
2 orange
Name: D, dtype: object
在这个示例中,我们首先创建了一个包含4列的DataFrame,然后使用loc
方法来比较A、B和C列的值,并在匹配时返回D列的值。在loc
方法中,我们使用布尔索引来筛选出匹配条件的行,并通过指定'D'列来获取对应的值。最后,我们打印出结果。