We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6f85247 commit 2de1791Copy full SHA for 2de1791
bin/filter_best_bitscore.py
@@ -7,14 +7,14 @@
7
import pandas as pd
8
9
10
-def filter_best_bitscore(df, group_col, score_col):
+def filter_dataframe(df, group_col, score_col):
11
idxmax = df.groupby(group_col)[score_col].idxmax()
12
df = df.loc[idxmax].reset_index(drop=True)
13
return df
14
15
def main(args):
16
blast_df = pd.read_csv(args.input)
17
- filtered_df = filter_best_bitscore(blast_df, args.group_col, args.score_col)
+ filtered_df = filter_dataframe(blast_df, args.group_col, args.score_col)
18
filtered_df.to_csv(args.output, index=False)
19
20
0 commit comments