From 5c1db0ca5d58b519c3dcc25006e46fb66b968518 Mon Sep 17 00:00:00 2001 From: MaoXianxin Date: Wed, 23 Jun 2021 15:08:11 +0800 Subject: [PATCH] Filter items by programming language --- projectFilter.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 projectFilter.py diff --git a/projectFilter.py b/projectFilter.py new file mode 100644 index 0000000..5920edd --- /dev/null +++ b/projectFilter.py @@ -0,0 +1,30 @@ +import pandas as pd +import argparse + +# Filter items by programming language +# Just need to modify csv_path, result_name, language +# Just change the parameters here +parser = argparse.ArgumentParser() +parser.add_argument('--csv_path', default='0618.csv', type=str) +parser.add_argument('--sort_values', default='notification count', type=str) +parser.add_argument('--result_name', default='java_csv.csv', type=str) +parser.add_argument('--language', default='java', type=str) +args = parser.parse_args() + + +csv1 = pd.read_csv(args.csv_path) + + +for i in range(len(csv1['project path'])): + csv1['project path'][i] = csv1['project path'][i].lower() + + +java_list = [] +for i in range(len(csv1)): + if args.language in csv1['project path'][i]: + java_list.append(i) + + +java_csv = csv1.loc[java_list] +java_csv = java_csv.sort_values(args.sort_values, ascending=False) +java_csv.to_csv(args.result_name, index=False) \ No newline at end of file -- GitLab