1. 程式人生 > >python資料處理(csv->折線圖)

python資料處理(csv->折線圖)

highs_lows.py

# -*- coding: utf-8 -*-
"""
Created on Mon May 29 10:00:16 2017

@author: Administrator
"""

import csv


filename = '300001_D.csv'
with open(filename) as f:
    reader = csv.reader(f)
    header_row = next(reader)
print(header_row)



with open(filename) as f:
    reader = csv.reader(f)
    header_row = next(reader)
    for
index, column_header in enumerate(header_row): print(index, column_header) with open(filename) as f: reader = csv.reader(f) header_row = next(reader) highs = [] for row in reader: high=row[2] highs.append(high) print(highs)

highs_lows02.py

# -*- coding: utf-8 -*-
""" Created on Mon May 29 10:59:38 2017 @author: Administrator """ import csv from datetime import datetime from matplotlib import pyplot as plt filename = '300001_D.csv' with open(filename) as f: reader = csv.reader(f) header_row = next(reader) dates, highs = [],[] for row in reader: current_date = datetime.strptime(row[0
], "%Y-%m-%d") dates.append(current_date) high=row[2] highs.append(high) print(dates, highs) # 根據資料繪製圖形 fig = plt.figure(dpi=128, figsize=(10, 6)) plt.plot(dates, highs, c='red') # 設定圖形的格式 plt.title("Daily high stock, July 2014", fontsize=24) plt.xlabel('', fontsize=16) fig.autofmt_xdate() plt.ylabel("Number (F)", fontsize=16) plt.tick_params(axis='both', which='major', labelsize=16) plt.show()

highs_lows03.py

# -*- coding: utf-8 -*-
"""
Created on Mon May 29 11:08:22 2017

@author: Administrator
"""

import csv
from datetime import datetime
from matplotlib import pyplot as plt

filename = '300001_D.csv'        
with open(filename) as f:
    reader = csv.reader(f)
    header_row = next(reader)
    dates, highs, lows = [],[],[]
    for row in reader:
        try:
            current_date = datetime.strptime(row[0], "%Y-%m-%d")
            high=row[2]
            low = row[4]
        except ValueError:
            print(current_date, 'missing data')
        else:
            dates.append(current_date)
            highs.append(high)
            lows.append(low)

    print(dates, highs, lows)



# 根據資料繪製圖形
fig = plt.figure(dpi=128, figsize=(10, 6))
#處的實參alpha 指定顏色的透明度。 Alpha 值為0表示完全透明, 
#1( 預設設定) 表示完全不透明
plt.plot(dates, highs, c='red', alpha=0.5)
plt.plot(dates, lows, c='blue', alpha=0.5)

# 設定圖形的格式
plt.title("Daily high and low Stock, 2017-(01,05)", fontsize=24)
plt.xlabel('', fontsize=16)
fig.autofmt_xdate()
plt.ylabel("Number(F)", fontsize=16)
plt.tick_params(axis='both', which='major', labelsize=16)
plt.show()

輸出圖片:
這裡寫圖片描述

學習《python程式設計:從入門到實踐》