在 pandas 中使用 for 循环从两列中附加最高值的列表

Question

我有两列：A 列和 B 列。

我想找出 A 列每一行的值是否大于 B 列同一行的值，如果是 append 则列出这些值。

如果 A 列中的值高于设定值，我可以 append 列表，但我不确定如何将其与 B 列中的值进行比较。

如果 A 列中的值高于 4，则下面的代码会附加列表。希望我走在正确的轨道上，可以用其他代码替换 4 吗？

list = []

for x in A:
    if x > 4:
        list.append(x)

print(list)

任何帮助将不胜感激。

谢谢！

Answer 1

一种方法可以是：

import pandas as pd

df = pd.DataFrame({"A":[2, 3, 4, 5], "B":[1, 4, 6, 3]}) # Test DataFrame

print(list(df[df["A"] > df["B"]]["A"]))

OUTPUT

[2, 5]