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

mac2026-02-05  0

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

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

代码过于简单,大佬勿喷


材料准备:

1.hs300.csv文件

步骤说明:

从 hs300.csv 读取数据,绘制 ma5 和 ma13 两条均线将数据存于 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 <class 'pandas.core.series.Series'> 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读写中的坑。

如有大佬,请勿喷。

最新回复(0)