1 Star 1 Fork 3

zhangdashenqi/the_zen_of_stata

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
1_data_in_Python_and_Stata.do 1.99 KB
一键复制 编辑 原始数据 按行查看 历史
zhangdashenqi 提交于 2019-09-28 19:38 +08:00 . update all file
************************************************
**** Stata之禅 ****
**** the Zen of Stata ****
************************************************
* Python与Stata的数据交互
python: print("Hello World")
python:
for i in range(1, 10):
for j in range(1, i+1):
print("%d*%d=%-4.0d" % (j, i, i*j), end="")
print("")
end
*-1.获取Tushare存款利率数据
clear
python:
from sfi import Data
import tushare as ts
# 创建数据集和变量
df = ts.get_deposit_rate()
Data.addObs(df.shape[0])
Data.addVarStrL('date')
Data.addVarStrL('deposit_type')
Data.addVarStrL('rate')
# 存储数据
Data.store('date', None, df.iloc[:,0])
Data.store('deposit_type', None, df['deposit_type'])
Data.store('rate', None, df['rate'])
end
br
*-2.获取电影票房数据
clear
python:
from sfi import Data
import tushare as ts
# 创建数据集和变量
df = ts.day_boxoffice()
Data.addObs(df.shape[0])
for i in range(0, df.shape[1]):
Data.addVarStrL(df.iloc[:, i].name)
# 存储数据
for i in range(0, df.shape[1]):
Data.store(df.iloc[:, i].name, None, df.iloc[:, i])
end
br
*-3.获取Tushare股票数据
clear
python:
from sfi import Data
import matplotlib.pyplot as plt
import numpy as np
import tushare as ts
# 定义数据类型
intList = [np.int8, np.int16, np.int32, np.int64,
np.uint8, np.uint16, np.uint32, np.uint64]
floatList = [np.float16, np.float32, np.float64]
# 获取数据
df = ts.get_hist_data('hs300')
df = df.reset_index() # 将索引转化为列
# 添加变量
Data.addObs(df.shape[0])
for i in range(0, df.shape[1]):
if df.iloc[:, i].dtype in intList:
Data.addVarInt(df.iloc[:, i].name)
elif df.iloc[:, i].dtype in floatList:
Data.addVarFloat(df.iloc[:, i].name)
else:
Data.addVarStrL(df.iloc[:, i].name)
# 存储数据
for i in range(0, df.shape[1]):
Data.store(df.iloc[:, i].name, None, df.iloc[:, i])
#绘制图形
df[['close', 'ma5', 'ma10', 'ma20']].plot(
kind='line',
figsize=(16, 9),
grid=True,
title='HS300',
fontsize=12)
plt.show()
end
br
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/zhangdashenqi/the_zen_of_stata.git
git@gitee.com:zhangdashenqi/the_zen_of_stata.git
zhangdashenqi
the_zen_of_stata
the_zen_of_stata
master

搜索帮助