Spaces:
Sleeping
Sleeping
Create qa_csv.py
Browse files- 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
|