上市公司高送转预测

1.

# -*- coding: utf-8 -*-

'''读取“2018年股票日交易数据.xlsx"，其字段名称如下：

Stkcd（股票代码）、Trddt（交易日期）、Clsprc（收盘价），

设计一个简单算法，用于检测上市公司在2017年度报告中是否有“高送转”行为，

同时获得上市公司高送转标识数据，其中有高送转行为的记为1，否则为0。

返回结果用一个数据框B1来表示，其中列名分别为a_code（股票代码）、b（高送转标识：1-有高送转，0-无高送转）

说明：考虑日收盘价格跌幅在35%以上的，视为该股票存在高送转行为，其跌幅是由于股票价格除权所致

'''

def return_values():

    import pandas as pd

    A=pd.read_excel('2018年股票日交易数据.xlsx')

    code=list(A['Stkcd'].unique())

    code_list=[]

    z_list=[]

    for i in range(len(code)):

        temp=A.iloc[A['Stkcd'].values==code[i],:]

        temp=temp.sort_values('Trddt')

        p1=temp['Clsprc'].values[:-1]

        p2=temp['Clsprc'].values[1:]

        t=(p2-p1)/p1

        if len(t[t<-0.35])>0:

            code_list.append(code[i])

            z_list.append(1)

        else:

            code_list.append(code[i])

            z_list.append(0)

    D1={'a_code':code_list,'b':z_list}

    B1=pd.DataFrame