提交 5c1db0ca 编写于 作者: M MaoXianxin

Filter items by programming language

上级 325a27d9
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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册