Python#常用的模塊和簡單用法

random 隨機模塊

import random

code = random.choice(stock_list) # 從一個列表中隨機選取元素下面是我目前經常用到的模塊，為了方便使用，不是有特殊需求的話，個人覺得一開始沒比亞每個模塊都很深入學習，只要知道自己常用的一些方法就行。隨時更新，請搜索使用。

random 隨機選取模塊：

import randoma = [1, 2, 3, 4, 5]print(random.choice(a)) # 隨機從列表中抽取一個元素code = random.choice(stock_list) # 從一個列表中隨機選取元素

os 文件夾模塊：

import os# 設置默認文件路徑os.chdir()os.chdir(uC:/Users/Ocean/OneDrive/class5/data/input_data/stock_data)df = pd.read_csv(sz300001.csv)print df

# 程序根目錄地址，os.pardir：父目錄 parent directory

root_path = os.path.abspath(os.path.join(current_file, os.pardir, os.pardir)) # 兩級父目錄print root_path

# 輸入數據根目錄地址

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))

# 在空白畫布上設置一塊區域

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 Axes3Dfig = plt.figure()ax = Axes3D(fig)ax.scatter(df[ma_long],df[ma_short],df[final_ratio], c=b) #繪製數據點# 設置坐標軸名字ax.set_zlabel(final_ratio) #坐標軸ax.set_ylabel(ma_short)ax.set_xlabel(ma_long)plt.show()

推薦閱讀：