1. 程式人生 > >python : pandas , matplotlib 批量繪圖

python : pandas , matplotlib 批量繪圖

for_roll_mean.py

# -*- coding: utf-8 -*-
import glob
import pandas as pd

import matplotlib.pyplot as plt
# 加這個兩句 可以顯示中文
plt.rcParams['font.sans-serif'] = [u'SimHei']
plt.rcParams['axes.unicode_minus'] = False

for f1 in glob.glob("*.csv"):
    df = pd.read_csv('f1, parse_dates=True, index_col=0 )
    fname =u'基金淨值圖'
    fcode = f1.split('.')[0]
    df['jz'].plot(figsize=(10,5),grid=True,legend='jz',label=fcode,title=fname)
    # 畫60日移動平均線  
    pd.rolling_mean(df['jz'], 60).plot(grid=True)
    f2 = fcode +'.png'
    print f2
    plt.savefig(f2)
    plt.close()    # 這句很必要
#