文件读写:把数据保存起来

💾 为什么要读写文件?

电脑关机后,内存里的数据就没了。要永久保存数据,就得存到文件里。

量化中常见的保存:

  • 📊 股票数据存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——处理数据的利器!

发表评论