只需要,添加一个 axis 参数即可。
if __name__ == "__main__":
def ff(x):
if x["1"] < 4:
return x["1"]
return x["2"]
pass
dd = pd.DataFrame()
df = pd.DataFrame({"1": [1, 2, 3,4], "2": [10, 11, 12,13]})
print(df)
print("---------------")
rr = df.apply(lambda x: ff(x), axis=1)
df["calc"] = rr
print(df)
pass
运行结果为:
1 2
0 1 10
1 2 11
2 3 12
3 4 13
---------------
1 2 calc
0 1 10 1
1 2 11 2
2 3 12 3
3 4 13 13