近日需要對excel的csv文件進行處理,求取某銀行歷年股價的均值方差等一系列數(shù)據(jù)
文件的構(gòu)成很簡單,部分如下所示
總共有接近七千行數(shù)據(jù),主要的工作就是將其中的股價數(shù)據(jù)提取出來,放入一個數(shù)組之中,然后利用numpy模塊即可求出需要的數(shù)據(jù)。
這里利用了csv模塊來對文件進行處理,最終實現(xiàn)的代碼如下:
import csv
import numpy as np
with open('pingan_stock.csv') as csv_file:
row = csv.reader(csv_file, delimiter=',')
next(row) # 讀取首行
price = [] # 建立一個數(shù)組來存儲股價數(shù)據(jù)
# 讀取除首行之后每一行的第二列數(shù)據(jù),并將其加入到數(shù)組price之中
for r in row:
price.append(float(r[1])) # 將字符串數(shù)據(jù)轉(zhuǎn)化為浮點型加入到數(shù)組之中
print(np.var(price)) # 輸出均值
print(np.mean(price)) # 輸出方差