Corin1998 commited on
Commit
eacb67f
·
verified ·
1 Parent(s): c204318

Create qa_csv.py

Browse files
Files changed (1) hide show
  1. export/qa_csv.py +13 -0
export/qa_csv.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import csv
2
+
3
+ def save_qa_csv(qa_list, links, path="data/qa_30.csv"):
4
+ # links: [(id, url)] →id→urlの辞書
5
+ linkmap = {i:u for i , u in links}
6
+ with open(path, "w", newline="", encoding="utf-8") as f:
7
+ w = csv.writer(f)
8
+ w.writerow(["q","a", "sources"])
9
+ for qa in qa_list:
10
+ ids = qa.get("sources", [])
11
+ urls = [linkmap.get(i, "") for i in ids]
12
+ w.writerow([qa.get("q",""), qa.get("a","")," ".joi(urls)])
13
+ return path