在 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]