繁体   English   中英

有没有更好的方法来基于多个条件从 pandas DataFrame 行 select 行?

[英]Is there a better way to select rows from a pandas DataFrame based on multiple conditions?

我正在尝试基于多个条件从 pandas DataFrame 中的 select 行。 代码如下所示:

row = videos_train_df[
             (videos_train_df['pid1']==pid1)
            &(videos_train_df['pid2']==pid2)
            &(videos_train_df['vid'] ==vid)]

有没有更好的方法(在代码可读性方面)做同样的事情?

您可以使用query

row = videos_train_df.query(
    f"pid1 == {pid1} and pid2 == {pid2} and vid == {vid}"
)

另请参阅问题。

我会在eq之后做all事情

new = df[df[['pid1','pid2','vid']].eq([pid1,pid2,vid]).all(1)]

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM