random 隨機選取模塊：
import random
a = [1, 2, 3, 4, 5]
print(random.choice(a))? # 隨機從列表中抽取一個元素
os 文件夾模塊：
import os
# 設置默認文件路徑
os.chdir()
os.chdir(u'C:/Users/Ocean/OneDrive/文檔/量化交易/量化小講堂/視頻課程/第五課資料/class5/data/input_data/stock_data')
df = pd.read_csv('sz300001.csv')
print df
# 獲取當前程序的地址
current_file = __file__
# 程序根目錄地址渠缕，os.pardir：父目錄 parent directory
root_path = os.path.abspath(os.path.join(current_file, os.pardir, os.pardir))? # 兩級父目錄
print root_path
# 輸入數(shù)據(jù)根目錄地址
input_data_path = os.path.abspath(os.path.join(root_path, 'data', 'input_data'))
time 時間模塊：
import time
獲取當前日期
date_now = time.strftime('%Y-%m-%d', time.localtime(time.time()))
計時器
start = time.time()
end = time.time()
used_time = str(end - start)
print "used_time: " + used_time
matplotlab.pyplot 作圖模塊
import matplotlib.pyplot as plt
# 添加空白畫布
fig = plt.figure(figsize=(12,5))
# 在空白畫布上設置一塊區(qū)域
ax = fig.add_subplot(1,1,1)
# 設置畫塊的標題
ax.set_title(str(code))
ax.set_xlabel('Time')? # 設置橫坐標x軸的名字
ax.set_ylabel('Return')? # 設置Y軸
# 畫一根2D線圖频伤，并設置名稱為'stock_return’
plt.plot(df[equity], label='stock_return')?
# 繪制散點圖
plt.scatter(df['ma_long'], df['final_ratio'], label='ma_long')
plt.legend(loc='best')? # 顯示圖線的名字
plt.show()? # 繪出圖像結果
mpl_toolkits.mplot3d? 繪制3D圖模塊
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = Axes3D(fig)
ax.scatter(df['ma_long'],df['ma_short'],df['final_ratio'], c='b') #繪制數(shù)據(jù)點
# 設置坐標軸名字
ax.set_zlabel('final_ratio') #坐標軸
ax.set_ylabel('ma_short')
ax.set_xlabel('ma_long')
plt.show()
random 隨機模塊
import random
code = random.choice(stock_list)? # 從一個列表中隨機選取元素
