009：获取20日均线数据

在《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线数据文件，打开即可，生成新的文件：