再《005》中我们获得了K线数据,现在我们要把他的20日均线数据也获取出来。然后通过计算后,保存在新的一列中:
import pandas as pd
import tkinter as tk
from tkinter import filedialog
import ospath=''def open_file():global pathpath = filedialog.askopenfilename(initialdir="./", title="Select file", filetypes=(("Text files", "*"), ("All files", "*.*")))print("Selected file:", path)root.quit() # 选择完文件后退出主循环root = tk.Tk()
root.title("File Selector")button = tk.Button(root, text="Open File", command=open_file)
button.pack()root.mainloop()# 读取表格数据
data = pd.read_csv(path, delimiter=',')# 读取数据文件
# df = pd.read_csv('0001.csv')]
df = pd.read_csv(path, sep=',')# 计算20日均线数据
df['20日均线'] = df['收盘'].rolling(window=20).mean()
df['20日均线'] = df['20日均线'].map('{:.2f}'.format) # 使用format()函数进行格式化# print(df)
filename = os.path.basename(path)df.to_csv('avg_20_'+filename, index=False)
运行:
选择之前下载的K线数据文件,打开即可,生成新的文件: