保留原本功能实现的情况下优化以下代码import random import pandas as pd import matplotlib.pyplot as plt fn = 'data.csv' products = ['商品1','商品2','商品3','商品4','商品5','商品6','商品7','商品8','商品9','商品10'] datelist = [] for month in range(1,13) : for day in range(1,32) : date = f'2019-{month:20d}-{day:02d}' datelist.append(date) datalist =[] for date in datelist : for it in products : sales = round(random.uniform(100,1000),2) datalist.append([date,it,sales]) df = pd.DataFrame(datalist, columns=['日期','商品名称','营业额']) df.to_csv('data.csv', index=False) df = pd.read_csv('data.csv') for product in df['products'].unique() : data = df.loc[df['products'] == product] plt.plot(data['date'],data['sales'],label=product) plt.xlabe1('Date') plt.ylabe1('sales') plt.title('Sales by Product') plt.legend() plt.show() df['month'] = pd.DatetimeIndex(df['date']).month groupeddata = df.groupby(['products','month'])['sales'].sum().unstack() groupeddata.plot(kind='bar') plt.xlabel('Products') plt.ylabel('Sales') plt.title('Sales by Month') plt.legend(title='Month',labels=['JAN','FEB','MAR','APR','MAY','JUN','JUL','AUG','SEP','OCT','NOV','DEV']) plt.show() df['quarter'] = pd.PeriodIndex(df['date'],freq='Q') groupeddata = df.groupby(['products','quarter'])['sales'].sum().unstack() groupeddata.plot(kind='pie',subplots=True) plt.title('Sales by Quarter') plt.legend(loc='center left',bbox_to_anchor=(1.0,0.5)) plt.show()
扫一扫
1033
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
