PYTHON与财务【下】--财务分析篇_python会计

方法一、PANDAS方法

# 获取会计科目数据
# 方法一、通过Pandas库从csv文件获取数据
# 写成函数，code为六位股票代码，itp为报表类型：'is for income_statement, bs for balance_sheet, cf for cash_flows
# n为默认参数，0表示最近一期，1表示最近两期，以下同

def select(code,itp,key,n=0):
    import pandas as pds
    s_file='/dev/shm/Finacials/{tp}{c}.csv'.format(c=code,tp=itp)
    df=pds.read_csv(s_file,encoding='gbk',index_col=0).dropna(axis=1)
    return df.ix[df.index.str.contains(key)].values[0][n]

%time select('600660','cf','经营活动产生的现金流量净额')

执行结果

方法二、直接处理csv文件

# 获取会计科目数据：方法二、直接处理文本
def select2(code,itp,key,n=0):
    s_file='/dev/shm/Finacials/{tp}{c}.csv'.format(c=code,tp=itp)
    ct=open(s_file,encoding='gbk').read().replace(