Tag: Python

通过Python下载中证行业分类文件

import requests

url = 'http://www.csindex.com.cn/uploads/downloads/other/files/zh_CN/ZzhyflWz.zip'
r = requests.get(url, stream = True)
with open("D:\\zhongzheng.zip", "wb") as Pypdf:
    for chunk in r.iter_content(chunk_size = 1024):
        if chunk:
            Pypdf.write(chunk)

通过Python下载网易金融数据文件

import requests

url = 'http://quotes.money.163.com/service/zycwzb_300525.html?type=report'
myfile = requests.get(url, allow_redirects=True)
open('D:\\300525.csv', 'wb').write(myfile.content)

解压缩zip文件

import zipfile

zip_file = zipfile.ZipFile("D:\\zhongzheng.zip")
print zip_file.namelist()
zip_file.extractall(path="D:\\stockdata")

安装

pip install matplotlib

绘制单条曲线

import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 2 * np.pi, 50)
plt.plot(x, np.sin(x)) # 如果没有第一个参数 x，图形的 x 坐标默认为数组的索引
plt.show() # 显示图形

输出如下

绘制两条曲线

import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 2 * np.pi, 50)
plt.plot(x, np.sin(x), x, np.sin(2*x))
plt.show()

输出如下

自定义曲线外观

import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 2 * np.pi, 50)
plt.plot(x, np.sin(x), 'r-.', x, np.cos(x), 'c--')
plt.show()

上述代码展示了两种不同的曲线样式：’r-.’ 和 ‘c–‘。字母 ‘r’ 和 ‘c’ 代表线条的颜色，后面的符号代表线和点标记的类型。

颜色： 蓝色 – ‘b’ ，绿色 – ‘g’ ，红色 – ‘r’ ，青色 – ‘c’， 品红 – ‘m’ ，黄色 – ‘y’，黑色 – ‘k’， 白色 – ‘w’

线： 直线 – ‘-‘ ，虚线 – ‘–‘ ，点线 – ‘:’ ，点划线 – ‘-.’

常用点标记：点 – ‘.’ ，像素 – ‘,’ ，圆 – ‘o’ ，方形 – ‘s’ ，三角形 – ‘^’

输出如下

增加标签和标题

import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 2 * np.pi, 50)
plt.plot(x, np.sin(x), 'r-x', label='Sin(x)')
plt.plot(x, np.cos(x), 'g-^', label='Cos(x)')
plt.legend() # 展示图例
plt.xlabel('Rads') # 给 x 轴添加标签
plt.ylabel('Amplitude') # 给 y 轴添加标签
plt.title('Sin and Cos Waves') # 添加图形标题
plt.show()

输出结果如下

绘制子图

x=np.linspace(0,2*np.pi,50)
plt.subplot(2,1,1) #（行，列，活跃位置）
plt.plot(x,np.sin(x),'r')
plt.subplot(2,1,2)
plt.plot(x,np.cos(x),'g')
plt.show()

散点图

x=np.linspace(0,2*np.pi,50)
y=np.sin(x)
plt.scatter(x,y)
plt.show()

使用Python非常优雅地实现买入和卖出股票的计算，如下。

stocks = set(context.stocks) 

holdings = set(get_holdings(context)) 

to_buy = stocks - holdings 

to_sell = holdings - stocks

利用集合的+ -法可以很方便计算当前持仓与目标持仓间需要买入和卖出的股票。

集合是无序无重复的元素集，不支持索引和切片，重复值自动剔除。集合支持 x in set, len(set),和 for x in set等操作。

创建集合

1）b=set([6,3,9,15,21]) #set()方法创建

2）a = {‘a’,’i’}  #{}方式创建，但不能为空，否则会默认创建字典

增加集合元素

1）add方法。b.add(45) #add方法支持增加单个元素

2）update方法。b.update(a) #update方法支持集合合并

删除集合元素

b.remove(9) #原集合删除了元素 9

集合的运算

a|b    #求a,b的并集

a&b    #求a,b的交集

a-b    #元素在a中，不在b中

a^b    #集合a除去a&b的交集元素

lambda

用法示例：
key=lambda x: abs(x)

相当于传入参数 x，输出结果 abs(x)

max与min

用法示例：
price = {
‘A’:123,
‘B’:450.1,
‘C’:12,
‘E’:444,
}
max_price=max(price.items(),key=lambda x:x[1])

对price.items()的序列求最大值，判断的标准是序列单元的第二个值

sorted

用法示例：
price = {
‘A’:123,
‘B’:450.1,
‘C’:12,
‘E’:444,
}
sorted_price=sorted(price.items(),key=lambda x:x[1],reverse=True)
对price.items()的序列排序，排序的标准是序列单元的第二个值，默认从小到大排序，reverse后从大到小排列。