再《005》中我们获得了K线数据,现在我们要把他的20日均线数据也获取出来。然后通过计算后,保存在新的一列中:
import pandas as pd
import tkinter as tk
from tkinter import filedialog
import os
path=''
def open_file():
global path
path = 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线数据文件,打开即可,生成新的文件: