使用 if 条件创建新的 pandas 数据框

Question

我有一个包含日期的熊猫股票价格数据框。 现在我尝试创建一个新的数据框，仅包含每月特定日期的值，如果这一天不存在，则使用另一个带有 if 和 elif 条件的数据框。

我使用了以下代码，但收到以下错误： “ValueError：DataFrame的真值不明确。使用a.empty、a.bool()、a.item()、a.any()或a.all()。”

for i in month:
    if df[df['month']==i] and df[df['day']==5]:
        new_df=df.series
    elif df[df['month']==i] and df[df['day']==6]:
        new_df=df.series

你知道如何解决这个问题吗？ 已经感谢您的帮助。

最好的问候， 马文

Answer 1

使用这些数据：

    month   day
0   1       5
1   1       6
2   3       5
3   4       5
4   5       6

那么你要做的就是：

temp_df = df.loc[df["day"].isin([5, 6])]
temp_df = temp_df.drop_duplicates(["month"], keep="first")
temp_df

    month   day
0   1       5
2   3       5
3   4       5  
4   5       6

如果您的数据未排序，您始终可以使用 df.sort_values 这样就可以了。