[pandas] groupby & matplot

groupby를 활용하여 matplot에 chart 그리기

GitHub

https://github.com/yiwonjae/Project_Python_Lib/blob/master/PandasMy/PandasMyLibs.py

'''
Groupby 된 DataFame을 chart로 구성

'''
def makeGroup_s003(df:DataFrame)->None:

    print(df.info())


    grouped:DataFrameGroupBy = df.groupby(['file_name'])

    import matplotlib
    matplotlib.use('TkAgg')
    import matplotlib.pyplot as plt

    for name, group in grouped:

        group:DataFrame = group

        group['sin'].plot()

    plt.show()
    

'''
row별 평균, groupby 사용
'''
def makeGroup_s004(df:DataFrame)->None:

    print(df.info())

    grouped:DataFrameGroupBy = df.groupby(['order'])

    import matplotlib
    matplotlib.use('TkAgg')
    import matplotlib.pyplot as plt

    datas=[]
    for name, group in grouped:

        group:DataFrame = group

        print('name: {0}, mean: {1}'.format(name, group['sin'].mean()))

        datas.append((name, group['sin'].mean()))

    merge = pd.DataFrame(datas, columns=['order', 'sin'])

    print(merge.info())
    print(merge)

    grouped_1:DataFrameGroupBy = df.groupby(['file_name'])

    for name, group in grouped_1:
        plt.plot(range(len(group['sin'])), group['sin'], 'r-')
        plt.plot(range(len(merge['sin'])), merge['sin'], 'b-')
        plt.show()