Update popularity.py

qqhsx · May 25, 2024 · 9048cb5 · 9048cb5
1 parent 04feef3
commit 9048cb5
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/popularity.py b/popularity.py
@@ -1,4 +1,12 @@
 import pywencai
-res = pywencai.get(question='退市股票', loop=True)
-res.to_csv('问财.csv', index=False, encoding='utf-8-sig')
+
+# 定义变量
+query = '退市股票'
+filename = f'{query}.csv'
+
+# 使用变量进行查询和保存结果
+res = pywencai.get(question=query, loop=True)
+res.to_csv(filename, index=False, encoding='utf-8-sig')
+
+# 打印结果
 print(res)