python - Pandas Dataframes - 根据 df1 中的条件填充 df2

Question

我有这个数据框 (df1)，我想将数据输入到另一个数据框 (df2) 中。如果 df1 的值大于 54，我希望 df2 中的同一行在“买入”列下为“买入”，如果不是，我希望它在“卖出”列下为“卖出”。我知道这听起来很简单，但出于某种原因，当我使用下面的代码执行此操作时，它会根据 df1 中的最后一个值设置 df2 中的所有值。

for x in df1['A']:
    if x > 54:
       df2['Buy'] = "Buy"

    else:
       df2['Sell'] = "Sell"

df1:

    Date
    2011-08-26     53.024284
    2011-08-29     55.454285
    2011-08-30     55.464287
    2011-08-31     55.795715
    2011-09-01     55.117142
    2011-09-02     53.534286

df2:

            Buy  Hold  Sell
Date
2011-08-26  0.0    0.0   0.0
2011-08-29  0.0    0.0   0.0
2011-08-30  0.0    0.0   0.0
2011-08-31  0.0    0.0   0.0
2011-09-01  0.0    0.0   0.0
2011-09-02  0.0    0.0   0.0

Answer 1

首先需要两个索引相同，然后可以在另一个 DataFrame df2 中使用由 df1 中的条件创建的 bool 掩码:

m = df1['A'] > 54
df2['Buy'] = df2['Buy'].mask(m, "Buy")
df2['Sell'] = df2['Sell'].mask(~m, "Sell")

与assign相同:

df2 = df2.assign(Buy= df2['Buy'].mask(m, "Buy"),Sell = df2['Sell'].mask(~m, "Sell"))

或者:

df2.loc[m, 'Buy'] = "Buy"
df2.loc[~m, 'Sell'] = "Sell"

---

print (df2)
            Buy  Hold  Sell
Date                       
2011-08-26    0   0.0  Sell
2011-08-29  Buy   0.0     0
2011-08-30  Buy   0.0     0
2011-08-31  Buy   0.0     0
2011-09-01  Buy   0.0     0
2011-09-02    0   0.0  Sell

如果索引不同使用reindex :

m = (df1['A'] > 54).reindex(df2.index, fill_value=False)