python数据可视化1

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from datetime import datetime


df = pd.read_csv("wm_viomall.csv") #读入数据

df['create_date'] = pd.to_datetime(df['create_date']) #将数据类型转换成datatime类型
df = df.set_index(['create_date'])  #将create_date设为索引

print(df.dtypes)
dataset = np.zeros((8,3))
for i in range(1,8):
    df_new = df['2017-0'+ str(i)]  #读取某月的数据
    count = len(df_new)   #计算某月行数
    price = df_new['price'].values  #读取某月价格
    if len(price):
        avg = np.average(price)
    else:
        avg = 0
    dataset[i][0] = i
    dataset[i][1] =count
    dataset[i][2] = avg
p1 =plt.plot(dataset[:,0] , dataset[:,1],'r', label = "sales")
p2 = plt.plot(dataset[:,0] , dataset[:,2], label = "price")
plt.grid(True)
plt.xlabel("month")
plt.legend(loc = 2)
plt.show()