我从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
编辑:我需要一个适用于所有情况的解决方案。不仅仅是不平等的论点。我用这个粗略的例子,但我需要抓住特定的任意行,否则彼此没有关系,等等。
答案 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