import csv
import os
import random
from matplotlib import markers
import matplotlib.pyplot as plt
import pandas as pd
if not os.path.exists("data.csv"):
#创建存储csv文件存储数据
file = open('data.csv', "w", encoding="utf-8-sig",newline='')
csv_head = csv.writer(file)
#表头
header = ['date','price']
csv_head.writerow(header)
file.close()
for i in range(12):
base = 1.5
base_add = base + 0.003 * i
r = 0
if i != 0:
r = random.random() * 0.03 - 0.01
print(r)
price = base_add + r
with open('data.csv', 'a+', encoding='utf-8-sig') as f:
f.write(str(i+1) + ',' + str(price) + '\n')
df = pd.read_csv('data.csv', encoding='utf-8-sig',usecols=["date", "price"])
date = df['date'].values + 1
price = df['price'].values * 10000
plt.plot(date,price,marker='o', color='r')
plt.xlabel('月份')
plt.ylabel('成交均价(元)')
plt.title('某市二手房屋历史均价走势图')
plt.legend()
plt.rcParams['font.sans-serif'] = ['SimSun']
plt.show()