快捷導(dǎo)航

Python常用的模塊和簡單用法

更新時間：2021年10月07日 13:33:20 作者：刀客特鹿

這篇文章主要給大家介紹Python#常用的模塊和簡單用法，以random 隨機(jī)模塊展開話題，感興趣的小伙伴可以參考一下

1、random 隨機(jī)模塊

import random

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

random 隨機(jī)選取模塊：

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

os 文件夾模塊：

import os

# 設(shè)置默認(rèn)文件路徑
os.chdir()
os.chdir(u'C:/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

輸入數(shù)據(jù)根目錄地址

input_data_path = os.path.abspath(os.path.join(root_path, 'data', 'input_data'))

time 時間模塊：

import time

獲取當(dāng)前日期

date_now = time.strftime('%Y-%m-%d', time.localtime(time.time()))

計(jì)時器

start = time.time()
end = time.time()
used_time = str(end - start)
print "used_time: " + used_time

2、matplotlab.pyplot 作圖模塊

import matplotlib.pyplot as plt

添加空白畫布

fig = plt.figure(figsize=(12,5))

在空白畫布上設(shè)置一塊區(qū)域

ax = fig.add_subplot(1,1,1)

設(shè)置畫塊的標(biāo)題

ax.set_title(str(code))
ax.set_xlabel('Time')  # 設(shè)置橫坐標(biāo)x軸的名字
ax.set_ylabel('Return')  # 設(shè)置Y軸

畫一根2D線圖，并設(shè)置名稱為 'stock_return'

plt.plot(df[equity], label='stock_return')

繪制散點(diǎn)圖

plt.scatter(df['ma_long'], df['final_ratio'], label='ma_long')

還有更多的圖形可以繪制，如果真的有需要，可以網(wǎng)上再搜索

plt.legend(loc='best')  # 顯示圖線的名字
plt.show()  # 繪出圖像結(jié)果

3、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ù)點(diǎn)
# 設(shè)置坐標(biāo)軸名字
ax.set_zlabel('final_ratio') #坐標(biāo)軸
ax.set_ylabel('ma_short')
ax.set_xlabel('ma_long')
plt.show()

您可能感興趣的文章: