如何将任意行附加到pandas python中的新数据框中?

时间:2018-02-06 05:03:42

标签: python pandas

我从CSV文件中读取数据框

df = pd.read_csv("data.csv") df

Match   teamA  teamB
0       3      2
1       4      3
2       2      4
3       4      2
4       3      4

现在我只想要第0,1,3行,因为这些是teamA赢得的行。 我想知道如何获取这些行并轻松将它们放入新的DF中。像

newdf = df(row0,row1,row3)

Match   teamA  teamB
0       3      2
1       4      3
3       4      2
编辑:我需要一个适用于所有情况的解决方案。不仅仅是不平等的论点。我用这个粗略的例子,但我需要抓住特定的任意行,否则彼此没有关系,等等。

1 个答案:

答案 0 :(得分:1)

只需使用布尔索引来选择teamA获胜时的行,即teamA> teamB:

aws --profile=sandbox s3 cp --recursive local_path bucket 

根据行号选择任意行(例如0和2):

>>> df[df['teamA'].gt(df['teamB'])]  # `gt` = greater than
       teamA  teamB
Match              
0          3      2
1          4      3
3          4      2