💾 为什么要读写文件?
电脑关机后,内存里的数据就没了。要永久保存数据,就得存到文件里。
量化中常见的保存:
- 📊 股票数据存CSV
- 📈 策略参数存JSON
- 📝 交易记录存Excel
👨💻 写文件
# 写入文本
with open("test.txt", "w") as f:
f.write("Hello量化!\n")
f.write("第二行")
👨💻 读文件
# 读取文本
with open("test.txt", "r") as f:
content = f.read()
print(content)
📊 存股票数据(CSV)
import pandas as pd
# 假设有股票数据
data = {
"日期": ["2024-01-01", "2024-01-02"],
"收盘": [100, 105],
"代码": ["600519", "600519"]
}
df = pd.DataFrame(data)
# 保存到CSV
df.to_csv("stock_data.csv", index=False, encoding="utf-8-sig")
print("保存成功!")
📖 读股票数据
# 从CSV读取
df = pd.read_csv("stock_data.csv")
print(df)
💡 量化实战:保存历史数据
import akshare as ak
import os
# 获取茅台数据
df = ak.stock_zh_a_hist(symbol="600519")
# 创建data文件夹
os.makedirs("data", exist_ok=True)
# 保存
filename = "data/600519.csv"
df.to_csv(filename, index=False, encoding="utf-8-sig")
print(f"已保存到 {filename}")
⚠️ 注意事项
- 文件路径要正确
- 中文字符用 utf-8-sig 编码
- index=False 不保存行号
📚 下一课
学会了文件读写,我们来学Pandas——处理数据的利器!