以下是一个使用Python的示例代码,用于比较列中的值,并根据列中的内容返回一个值:
import pandas as pd
# 创建一个包含“比较列”的DataFrame
df = pd.DataFrame({'列1': [1, 2, 3, 4, 5],
'比较列': ['A', 'B', 'A', 'B', 'A']})
# 定义一个函数,根据比较列的内容返回相应的值
def return_value(row):
if row['比较列'] == 'A':
return '值A'
elif row['比较列'] == 'B':
return '值B'
else:
return '其他值'
# 使用apply函数将函数应用到DataFrame的每一行,并创建一个新的列来存储返回的值
df['返回值'] = df.apply(return_value, axis=1)
# 打印结果
print(df)
运行以上代码,将输出以下结果:
列1 比较列 返回值
0 1 A 值A
1 2 B 值B
2 3 A 值A
3 4 B 值B
4 5 A 值A
在这个示例中,我们创建了一个包含两列的DataFrame,其中一列是“比较列”。然后我们定义了一个函数return_value
,该函数会根据比较列的内容返回相应的值。我们使用apply
函数将该函数应用到DataFrame的每一行,并创建一个新的列来存储返回的值。最后,我们打印出DataFrame的结果。
上一篇:比较列中的整数,输出结果错误。