数学建模处理数据

import pandas as pd
import numpy as np
d1 = []
df4 = []
a = []
b = []
i = 0
data = pd.read_excel('jijin.xlsx', header=0)
d5 = data['NAV']
with open('sharp.txt','r') as f:
    lines = f.readlines()

def statistics(f,ord,data):
    s = f
    x = ord
    while f == s:
        if ord == len(data['Symbol']):
            return ord-x
        f = data['Symbol'][ord]
        ord += 1
    return ord-1-x

while i < len(data['Symbol']):
    num = statistics(data['Symbol'][i],i,data)
    print 'num : ',num
    df4.append(num)
    j = 0
    m = 0
    while j< 3:
        a.append(0)
        j += 1
    while m < 3:
        b.append(0)
        m += 1
    i += 3
    s = i
    while i < s+num - 6:
        d2 = []
        for x in [-3,-2,-1,0,1,2,3]:
            d2.append(d5[i+x])
        d2 = np.array(d2)
        d3 = np.max(d2)
        d4 = np.min(d2)
        if d3 == d5[i] or d4 == d5[i]:
            a.append(1),b.append(0)
        else:
            a.append(0),b.append(1)
        i +=1
    j = 0
    m = 0
    while j< 3:
        a.append(0)
        j += 1
    while m < 3:
        b.append(0)
        m += 1
    i +=3

def search(s,lines):
    for da in lines:
        d = da.split(',')
        i = int(d[0])
        if s == i:
            return d[1]

s = 0
for c in data['Symbol']:
    if c == s:
        d1.append(float(e))
    else:
        e = search(c,lines)
        d1.append(float(e))
        s = c
l = len(df4)
df1 = pd.DataFrame(a,index = np.arange(len(data['Symbol'])),columns = ['a'] )
df2 = pd.DataFrame(b,index = np.arange(len(data['Symbol'])),columns = ['b'] )
df3 = pd.DataFrame(d1,index = np.arange(len(data['Symbol'])),columns = ['sharp'] )
df4 = pd.DataFrame(df4,index = np.arange(l),columns = ['numbers'] )
data.insert(5,'a',df1)
data.insert(6,'b',df2)
data.insert(7,'sharp',df3)
data.insert(8,'numbers',df4)
data.to_excel('test_add10.xlsx')
  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值