zoukankan      html  css  js  c++  java
  • python pandas模

    介绍

      pandas是一个强大的Python数据分析的工具包。
      pandas是基于NumPy构建的。
      pandas的主要功能
        具备对其功能的数据结构DataFrame、Series
        集成时间序列功能
        提供丰富的数学运算和操作
        灵活处理缺失数据
      安装方法:pip install pandas
      引用方法:import pandas as pd

    pandas:Series

      Series是一种类似于一位数组的对象,由一组数据和一组与之相关的数据标签(索引)组成。
      Series比较像列表(数组)和字典的结合体
      创建方式:

        pd.Series([4,7,-5,3])
        pd.Series([4,7,-5,3],index=['a','b','c','d'])
        pd.Series({'a':1, 'b':2})
        pd.Series(0, index=['a','b','c','d'])
      获取值数组和索引数组:values属性和index属性

    pandas:Series特性
      Series支持NumPy模块的特性(下标):
        从ndarray创建Series:Series(arr)
        与标量运算:sr*2
        两个Series运算:sr1+sr2 #index跟index运算
        索引:sr[0], sr[[1,2,4]]
        切片:sr[0:2](切片依然是视图形式)
        通用函数:np.abs(sr)
        布尔值过滤:sr[sr>0]
        统计函数:mean() sum() cumsum()

      Series支持字典的特性(标签):
        从字典创建Series:Series(dic),
        in运算:’a’ in sr、for x in sr
        键索引:sr['a'], sr[['a', 'b', 'd']]
        键切片:sr['a':'c']
        其他函数:get('a', default=0)等

    整数索引
      整数索引的pandas对象往往会使新手抓狂。
      例:
        sr = np.Series(np.arange(4.))
        sr[-1]
      如果索引是整数类型,则根据整数进行数据操作时总是面向标签的。
        loc属性 以标签解释
        iloc属性 以下标解释

    数据对齐
      pandas在运算时,会按索引进行对齐然后计算。如果存在不同的索引,则结果的索引是两个操作数索引的并集。
      例:
      sr1 = pd.Series([12,23,34], index=['c','a','d'])
      sr2 = pd.Series([11,20,10], index=['d','c','a',])
      sr1+sr2
      sr3 = pd.Series([11,20,10,14], index=['d','c','a','b'])
      sr1+sr3
      # 没有相同索引的显示NaN,缺失值是NaN

      如何在两个Series对象相加时将缺失值设为0?
      sr1.add(sr2, fill_value=0)
      灵活的算术方法:add, sub, div, mul

    series缺失填充
      缺失数据:使用NaN(Not a Number)来表示缺失数据。其值等于np.nan。内置的None值也会被当做NaN处理。
      处理缺失数据的相关方法:
        dropna() 过滤掉值为NaN的行
        fillna() 填充缺失数据
        isnull() 返回布尔数组,缺失值对应为True
        notnull() 返回布尔数组,缺失值对应为False
      过滤缺失数据:sr.dropna() 或 sr[data.notnull()]
      填充缺失数据:fillna(0)

    DataFrame
      DataFrame是一个表格型的数据结构,含有一组有序的列。
      DataFrame可以被看做是由Series组成的字典,并且共用一个索引。
      创建方式:
        pd.DataFrame({'one':[1,2,3,4],'two':[4,3,2,1]})
        pd.DataFrame({'one':pd.Series([1,2,3],index=['a','b','c']), 'two':pd.Series([1,2,3,4],index=['b','a','c','d'])})
        ……

      DataFrame一般导入数据来用
      csv文件读取与写入:
        df.read_csv('filename.csv')
        df.to_csv()

    DataFrame查看数据
      查看数据常用属性及方法:
        index 获取索引
        T 转置
        columns 获取列索引
        values 获取值数组
        describe() 获取快速统计

      DataFrame各列name属性:列名
        rename(columns={'name': 'newname'})

    DataFrame索引和切片
      DataFrame有行索引和列索引。
      通过标签获取:
        df['A'] #A字段所有数据
        df[['A', 'B']] #A、B字段所有数据(多字段需要再加[])
        df['A'][0] #A字段的第0行
        df[0:10][['A', 'C']] #0到10行,字段A和B的数据(可以多个字段)

      推荐下面的方法
        df.loc[:,['A','B']]
        df.loc[:,'A':'C'] #所有行,字段A和B的数据(最多俩个字段)
        df.loc[0,'A'] #0行,A字段的数据
        df.loc[0:10,['A','C']] #0到10行,字段A和B的数据(可以多个字段)

    DataFrame索引和切片
      通过位置获取:
        df.iloc[3] #第三行(位置索引)
        df.iloc[3,3] #第三行的第三列的值
        df.iloc[0:3,4:6] #0到3行,4到6列的值
        df.iloc[1:5,:] #1到5行的所有值
        df.iloc[[1,2,4],[0,3]] #1、2、4行的0到3列的值

      通过布尔值过滤:
        df[df['A']>0] #显示A字段大于0的所有数据
        df[df['A'].isin([1,3,5])] #显示A字段在[1,3,5]的所有数据
        df[df<0] = 0 #?

    DataFrame数据对齐与缺失数据
      DataFrame对象在运算时,同样会进行数据对其,结果的行索引与列索引分别为两个操作数的行索引与列索引的并集。

      DataFrame处理缺失数据的方法:
        dropna(axis=0,how='any',…) #axis:1是列,0是行;how:all全部NaN删除,any有就删除
        fillna() #替换NaN
        isnull() #返回布尔
        notnull() #与上面相反

    pandas常用方法(适用Series和DataFrame):
      mean(axis=0,skipna=False)
      sum(axis=1)
      sort_index(axis, …, ascending) 按行或列索引排序
      sort_values(by, axis, ascending) 按值排序
      NumPy的通用函数同样适用于pandas

      apply(func, axis=0) 将自定义函数应用在各行或者各列上,func可返回标量或者Series
      applymap(func) 将函数应用在DataFrame各个元素上
      map(func) 将函数应用在Series各个元素上

    层次化索引:
      层次化索引是Pandas的一项重要功能,它使我们能够在一个轴上拥有多个索引级别。
      例:data=pd.Series(np.random.rand(9), index=[['a', 'a', 'a', 'b', 'b', 'b', 'c', 'c', 'c'], [1,2,3,1,2,3,1,2,3]])

    pandas时间对象:
      时间序列类型:
        时间戳:特定时刻
        固定时期:如2017年7月
        时间间隔:起始时间-结束时间
      Python标准库:datetime
        date time datetime timedelta
        dt.strftime()
        strptime()
      第三方包:dateutil
        dateutil.parser.parse()
      成组处理日期:pandas
        pd.to_datetime(['2001-01-01', '2002-02-02'])
      产生时间对象数组:date_range
        start 开始时间 string or datetime-like
        end 结束时间 string or datetime-like
        periods 时间长度 integer or None
        freq 时间频率,默认为'D',可选H(our),W(eek),B(usiness),S(emi-)M(onth),(min)T(es), S(econd), A(year),…

    pandas文件读取
      读取文件:从文件名、URL、文件对象中加载数据
        read_csv 默认分隔符为csv
        read_table 默认分隔符为
        read_excel 读取excel文件
      读取文件函数主要参数:
        sep 指定分隔符,可用正则表达式如's+'
        header=None 指定文件无列名
        names 指定列名
        index_col 指定某列作为索引
        skip_row 指定跳过某些行
        na_values 指定某些字符串表示缺失值
        parse_dates 指定某些列是否被解析为日期,布尔值或列表

    pandas写入数据
      写入到文件:
        to_csv
      写入文件函数的主要参数:
        sep
        na_rep 指定缺失值转换的字符串,默认为空字符串
        header=False 不输出列名一行
        index=False 不输出行索引一列
        cols 指定输出的列,传入列表

      其他文件类型:json, XML, HTML, 数据库
      pandas转换为二进制文件格式(pickle):
        save
        load

    Matplotlib:绘图和可视化

      Matplotlib是一个强大的Python绘图和数据可视化的工具包。

      安装方法:pip install matplotlib
      引用方法:import matplotlib.pyplot as plt

      绘图函数:plt.plot()
      绘图也可以直接使用pandas的数据:df[['ma5', 'ma10']].plot()
      显示图像:plt.show()

    plot函数:
      线型linestyle(-,-.,--,..)
      点型marker(v,^,s,*,H,+,x,D,o,…)
      颜色color(b,g,r,y,k,w,…)
      plot函数绘制多条曲线
      标题:title
      x轴:xlabel
      y轴:ylabel

    其他类型图像:
      hist 频数直方图

    Matplotlib:画布与图

      画布:figure
        fig = plt.figure()
      图:subplot
        ax1 = fig.add_subplot(2,2,1)
      调节子图间距:
        subplots_adjust(left, bottom, right, top, wspace, hspace)

    五日均线、十日均线求金叉死叉

    数据源是网上下载一段时间的股票信息

    import pandas as pd
    import numpy as np
    import matplotlib.pyplot as plt
    
    df = pd.read_csv('601318.csv', index_col='date', parse_dates=['date'])
    
    df['ma5'] = np.nan
    df['ma10'] = np.nan
    
    # 第一步计算ma
    # 循环计算,速度非常慢
    # for i in range(4, len(df)):
    #     df.loc[df.index[i], 'ma5'] = df['close'][i-4:i+1].mean()
    # for i in range(9, len(df)):
    #     df.loc[df.index[i], 'ma10'] = df['close'][i-9:i+1].mean()
    
    # 方案2:cumsum
    
    # close     =  [10, 11, 12, 13, 14, 15, 16]
    # close.cumsum=[10, 21, 33, 46, 60, 75, 91]
    #                                -   -   -
    #               [nan,nan,nan,nan,0,  10, 21, 33, 46, 60, 75, 91]
    
    # sr = df['close'].cumsum()
    # df['ma5'] = (sr - sr.shift(1).fillna(0).shift(4))/5
    # df['ma10'] = (sr - sr.shift(1).fillna(0).shift(9))/10
    
    # 方案3:rolling
    
    df['ma5'] = df['close'].rolling(5).mean()
    df['ma10'] = df['close'].rolling(10).mean()
    
    df = df.dropna()
    
    df[['ma5', 'ma10']].plot()
    plt.show()
    # 第二部 判断金叉死叉
    # 方案一
    # 金叉 短期<=长期 短期>长期
    # 死叉 短期>=长期 短期<长期
    # sr = df['ma5'] <= df['ma10']
    #
    # golden_cross = []
    # death_cross = []
    # for i in range(1, len(sr)):
    #     # if sr.iloc[i] == True and sr.iloc[i + 1] == False: 开始想的是加1,但是索引溢出
    #     if sr.iloc[i - 1] == True and sr.iloc[i] == False:
    #         golden_cross.append(sr.index[i])
    #     if sr.iloc[i - 1] == False and sr.iloc[i] == True:
    #         death_cross.append(sr.index[i])
    
    # 方案2
    
    golden_cross = df[(df['ma5'] <= df['ma10']) & (df['ma5'] > df['ma10']).shift(1)].index
    death_cross = df[(df['ma5'] >= df['ma10']) & (df['ma5'] < df['ma10']).shift(1)].index
  • 相关阅读:
    C语言寒假大作战03
    C语言寒假大作战02
    C语言寒假大作战01
    助教学期总结
    第十六周助教总结
    C语言I博客作业11
    第十五周助教总结
    C语言I博客作业10
    第十四周助教总结
    C语言寒假大作战03
  • 原文地址:https://www.cnblogs.com/wangyufu/p/7515107.html
Copyright © 2011-2022 走看看