# -*- coding: utf8 -*-
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib.pyplot import savefig
sales_df = pd.read_excel('F:/anaconda/a1.xlsx', names=["zaofa1", "zaofa2", "zaofa3", "zaofa4", "zaofa5"
,"zaofa6", "zaofa7", "zaofa8"])
#读取文件,并且为列取名字
aq = sales_df.ix[sales_df["zaofa8"] == "Barton LLC", ["zaofa8", "zaofa7", "zaofa5", "zaofa1"]]
#取出特定行与列,并且按照该次序排列
'''科学计算'''
aq["sum"] = ((aq["zaofa1"] * aq["zaofa5"]) - 12) *2 #增加列
aq["成绩"] = ((aq["zaofa1"] * aq["zaofa5"]) - 12) *3
# print(sales_df[sales_df.zaofa8.isin(["Barton LLC", "Fritsch, Russel and Anderson"])])
#zaofa8列含有"Barton LLC"与"Fritsch, Russel and Anderson"行
# w = sales_df[sales_df.zaofa8.isin(["Barton LLC", "Fritsch, Russel and Anderson"])]
#写入值
'''数据可视化'''
'''导入行列数据'''
width = sales_df.zaofa5
height = sales_df.zaofa1
# plt.style.use("dark_background") #设置背景样式
'''线形样式设置,颜色,各种参数直接在里面改'''
# plt.scatter(width, height) #散点图
plt.plot(width, height, color="y", marker="*") #折线图
# plt.barh(width, height) #水平条形图
# plt.bar(width, height, color="#008080") #条形图
'''图例、标题设置'''
plt.title("Lu zaofa made it!")
plt.xlabel("zaofa1")
plt.ylabel("zaofa2")
# plt.show()
savefig("F:/anaconda/picture/a598.jpg") #存储图片
writer = pd.ExcelWriter('F:/anaconda/sales_summary3101.xlsx', engine='xlsxwriter')
#储存到规定目录下,并命名工作簿
aq.to_excel(writer, 'zaofa', index=False) #为工作表起名
writer.save() #储存文件