以下是一个示例代码,说明如何比较多个列并获取最大值,并返回对应的特定值。
假设有一个包含三列的数据集,分别为A列、B列和C列。我们需要比较这三列的值,并返回对应的特定值。具体步骤如下:
def get_max_value(row):
if row['A'] > row['B'] and row['A'] > row['C']:
return '特定值A'
elif row['B'] > row['A'] and row['B'] > row['C']:
return '特定值B'
elif row['C'] > row['A'] and row['C'] > row['B']:
return '特定值C'
else:
return '无最大值'
import pandas as pd
# 创建示例数据集
data = {'A': [1, 2, 3, 4, 5],
'B': [2, 3, 4, 5, 6],
'C': [3, 4, 5, 6, 7]}
df = pd.DataFrame(data)
# 应用函数并创建新列
df['最大值对应特定值'] = df.apply(get_max_value, axis=1)
# 打印结果
print(df)
运行以上代码,输出结果如下:
A B C 最大值对应特定值
0 1 2 3 特定值C
1 2 3 4 特定值C
2 3 4 5 特定值C
3 4 5 6 特定值C
4 5 6 7 特定值C
可以看到,最大值对应的特定值被存储在了新的“最大值对应特定值”列中。
上一篇:比较多个列表的某些索引值
下一篇:比较多个目标文档和多个源文档