Python常用的模块和用法

random 随机模块

code = random.choice(stock_list) # 从一个列表中随机选取元素下面是我目前经常用到的模块

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

程序根目录地址，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()))
print(date_now) # 2019-03-23

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

matplotlab.pyplot 绘图模块

import matplotlib.pyplot as plt
from numpy import *

#添加空白画布
fig = plt.figure(figsize=(12,5))

#在空白画布上设置一块区域
ax = fig.add_subplot(1,1,1)

ax.set_title(str('title')) #设置画块的标题
ax.set_xlabel('Time')  # 设置横坐标x轴的名字
ax.set_ylabel('Return')  # 设置Y轴

#画一根2D线图，并设置名称为'line’
x = linspace(0, 5, 10)
y = x ** 2
plt.plot(x,y, label='line')

#绘制散点图
plt.scatter(x, y, label='ma_long')

plt.legend(loc='best')  # 显示图线的名字
plt.show()  # 绘出图像结果

mpl_toolkits.mplot3d 绘制3D图模块

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from numpy import *

fig = plt.figure()
ax = Axes3D(fig)

x = linspace(0, 5, 10)
y = x ** 2
ax.scatter(x,y,s=22**2, c='b') #绘制数据点

# 设置坐标轴名字
ax.set_zlabel('final_ratio') #坐标轴
ax.set_ylabel('ma_short')
ax.set_xlabel('ma_long')
plt.show()

Pygame

安装模块

$sudo pip3 install pygame

验证安装：

$python3 -m pygame.examples.aliens

Requests模块

安装：

$sudo pip3 install requests

请求URL：

无参GET请求:

 r=requests.get("http://pythontab.com/justTest")

带参GET请求：

payload = {'key1': 'value1', 'key2': 'value2'}
r = requests.get("http://pythontab.com/justTest", params=payload)
请求结果：
>>>print r.url
http://pythontab.com/justTest?key2=value2&key1=value1

POST请求：  
r = requests.post("http://pythontab.com/postTest", data = {"key":"value"})
以上得知，post请求参数是以data关键字参数来传递的。
现在的data参数传递的是字典，我们也可以传递一个json格式的数据，如下：
>>> import json
>>> import requests
>>> payload = {"key":"value"}
>>> r = requests.post("http://pythontab.com/postTest", data = json.dumps(payload))