Python简单教程:利用pandas对数据的操作(写入excel)

Python简单教程:利用pandas对数据的操作(写入excel)

因为之前在对Excel操作的时候，遇到了一些坑。估计小白们都会遇到，所以做了一个最简单的教程，可以为小白避免了很多坑。

代码过于简单，大佬勿喷

---

材料准备:

1.hs300.csv文件

步骤说明:

1. 从 hs300.csv 读取数据，绘制 ma5 和 ma13 两条均线
2. 将数据存于 excel 中，并在 excel 中也绘制相关线图

代码块:

1.导入所需的python模块

import matplotlib.pyplot as plt
import pandas as pd
import xlsxwriter

2.通过pandas模块获取相应列的值

data=pd.read_csv('hs300.csv')
ma5=data['ma5']  # 注意type 
ma13=data['ma13']

3.利用matplotlib.pyplot模块对数据进行绘制操作
(仅对数据进行简单操作,不涉及精细操作)

# 画图
plt.figure(1) 
# 共划分为3个区域
area1=plt.subplot(2,2,1)
area2=plt.subplot(2,2,2)
area3=plt.subplot(2,1,2)
plt.sca(area1)    # ma5区域1
plt.title('ma5')
ma5.plot()
plt.sca(area2)    # ma13区域2
plt.title('ma13')
ma13.plot()
plt.sca(area3)    # ma5和ma13区域3
ma5.plot()
ma13.plot()
# 将绘制的图另存为PNG图片
plt.savefig('data.png')
plt.show()

扩展:
(Series转DataFrame)

# ma= {'ma5':ma5.values,'ma13':ma13.values}
# data = pd.DataFrame(ma)
# data.to_excel('data.xlsx',sheet_name='data')

4.将所得数据写入excel,包括png图片
(简单遍历写入)

code=xlsxwriter.Workbook('data.xlsx')
content=code.add_worksheet("data")
for i in range(0,1):
    for j in range(0,len(ma5)):
        content.write(j,i,ma5[j])
        content.write(j,i+1,ma13[j])
content.insert_image(0,3,'data.png')
code.close()

操作结果展示:

data.png

data.xlsx

( 发现索引忘记写进去了，请自行解决)

总结:

仅对小白的简单教程，特别是excel的读写，数据的绘制图像，避免了excel读写中的坑。

如有大佬，请勿喷。