目录
二、用pandas读写excel文件(方便,但是还是要掌握上面的方法)
一、excel 文件
#示例2:将示例1读取的 Excel 文件内容,写入到另一个Excel中,对学校所在省份进行简单判断。第一行合并单元格显示标题。
#(1)导入模块:xlwt
import xlrd
import xlwt
#(2)读取文件内容
def read_excel(file_name): #定义读取文件函数
wb = xlrd.open_workbook(file_name) #创建读取文件的对象wb
sheet = wb.sheet_by_index(0)
schools = []
for row in range(sheet.nrows):
school = []
for col in range(sheet.ncols):
content = sheet.cell_value(row,col)
school.append(content)
schools.append(school)
return schools #提供返回值
#(3)写入文件内容
def write_excel(schools): #定义写入文件函数
#(2)构造工作簿:Workbook
wb = xlwt.Workbook(encoding = 'utf-8') #创建写入文件的对象wb
#(3)为工作簿添加表单:Worksheet
s = wb.add_sheet('上海市高校信息表') #创建一个表单
#(4)根据行列序号写入内容
#添加文本的样式(字体和对齐)
font = xlwt.Font() #字体
font.bold = True
font.height = 400
font.underline = True
font.colour_index = 6 #0:黑,1:白,2:红,3:绿,4:蓝,5:黄,6:紫
align = xlwt.Alignment() #对齐
align.horz = 2 #水平居中 1:左,2:中,3:右
align.vert = 1 #垂直居中 0:上,1:中,2:下
style = xlwt.XFStyle() #样式
style.font = font
style.alignment = align
s.write_merge(0,0,0,6,'上海市高校信息表',style) #写表标题并合并单元格(A1:A7)
for col in range(7): #写表列表名称
s.write(1,col,schools[0][col]) #第2行第1列开始写内容school[0][col]
row_num = 2 #从第3行开始写数据
for school in schools: #一行一行写数据内容
if school[2] == '上海市':
for col in range(7):
s.write(row_num,col,school[col])
row_num = row_num + 1
#(5)保存文件内容
wb.save('../R&Q_pic/上海市高校信息表.xls')
school_list = read_excel('../Stu_pack/wordcloud/school.xls') #调用读取函数,将素材里的文件school.xls作为实参传递给形参
write_excel(school_list) #调用写入函数,将读取的数据作为实参传递给形参写入到文件"上海市高校信息表.xls"里保存
schools_list = read_excel('../R&Q_pic/上海市高校信息表.xls')
i=1
for school in schools_list:
if i<13:
print(school)
i+=1
输出:
['上海市高校信息表', '', '', '', '', '', '']
['招生单位代码', '招生单位名称', '所在省份', '是否985', '是否211', '是否自主划线', '学校类型']
['10246', '复旦大学', '上海市', '是', '是', '是', '综合类']
['10247', '同济大学', '上海市', '是', '是', '是', '理工类']
['10248', '上海交通大学', '上海市', '是', '是', '是', '综合类']
['10251', '华东理工大学', '上海市', '否', '是', '否', '理工类']
['10252', '上海理工大学', '上海市', '否', '否', '否', '理工类']
['10254', '上海海事大学', '上海市', '否', '否', '否', '理工类']
['10255', '东华大学', '上海市', '否', '是', '否', '理工类']
['10256', '上海电力学院', '上海市', '否', '否', '否', '理工类']
['10259', '上海应用技术大学', '上海市', '否', '否', '否', '理工类']
['10264', '上海海洋大学', '上海市', '否', '否', '否', '农林类']
二、用pandas读写excel文件(方便,但是还是要掌握上面的方法)
(1)源码:
import pandas as pd
data=pd.read_excel('../Stu_pack/wordcloud/school.xls',index_col=0)#用pandas读取文件数据并创建文件对象
data
输出:
招生单位名称 | 所在省份 | 是否985 | 是否211 | 是否自主划线 | 学校类型 | |
---|---|---|---|---|---|---|
招生单位代码 | ||||||
10001 | 北京大学 | 北京市 | 是 | 是 | 是 | 综合类 |
10002 | 中国人民大学 | 北京市 | 是 | 是 | 是 | 综合类 |
10003 | 清华大学 | 北京市 | 是 | 是 | 是 | 理工类 |
10004 | 北京交通大学 | 北京市 | 否 | 是 | 否 | 理工类 |
10005 | 北京工业大学 | 北京市 | 否 | 是 | 否 | 理工类 |
... | ... | ... | ... | ... | ... | ... |
90114 | 北京系统工程研究所 | 北京市 | 否 | 否 | 否 | NaN |
90115 | 解放军医学院 | 北京市 | 否 | 否 | 否 | NaN |
90201 | 武警指挥学院 | 天津市 | 否 | 否 | 否 | NaN |
90202 | 武警工程大学 | 陕西省 | 否 | 否 | 否 | 军事类 |
90203 | 武警后勤学院 | 天津市 | 否 | 否 | 否 | NaN |
870 rows × 6 columns
(2)源码:
data.tail(10)
输出:
招生单位名称 | 所在省份 | 是否985 | 是否211 | 是否自主划线 | 学校类型 | |
---|---|---|---|---|---|---|
招生单位代码 | ||||||
90109 | 中国航天员科研训练中心 | 北京市 | 否 | 否 | 否 | NaN |
90110 | 北京跟踪与通信技术研究所 | 北京市 | 否 | 否 | 否 | NaN |
90111 | 中国国防科技信息中心 | 北京市 | 否 | 否 | 否 | NaN |
90112 | 西北核技术研究所 | 陕西省 | 否 | 否 | 否 | NaN |
90113 | 中国空气动力研究与发展中心 | 四川省 | 否 | 否 | 否 | NaN |
90114 | 北京系统工程研究所 | 北京市 | 否 | 否 | 否 | NaN |
90115 | 解放军医学院 | 北京市 | 否 | 否 | 否 | NaN |
90201 | 武警指挥学院 | 天津市 | 否 | 否 | 否 | NaN |
90202 | 武警工程大学 | 陕西省 | 否 | 否 | 否 | 军事类 |
90203 | 武警后勤学院 | 天津市 | 否 | 否 |
(3)源码:
data.count()
输出:
招生单位名称 870
所在省份 870
是否985 870
是否211 870
是否自主划线 870
学校类型 454
dtype: int64
(4)源码:
data=data[data.所在省份=='上海市']
data
data.to_excel('../R&Q_pic/上海市高校信息.xls')
三、词云库wordcloud的安装与应用
1、安装
pip install wordcloud #网络安装
python-m pip install 本地路径 #本地安装
2、应用
生成步骤:
创建词云对象-加载词云文本-输出词云图片(文件)
(1)默认的矩形词云图片
(2)提供的图形词云图片
(1)源码:
import wordcloud #导入库模块
from PIL import Image
wc=wordcloud.WordCloud(r'C:\Windows\Fonts\msyh.ttc') #创建词云对象
with open(r'C:\Users\Administrator\Desktop\202209408\Stu_pack\wordcloud\Dream It Possible.txt',encoding='utf-8')as file:
fr=file.read()
wc.generate(fr) #加载词云文本
wc.to_file('../R&Q_pic/test.png') #输出词云图片(文件)
Image.open('../R&Q_pic/test.png')
输出:
(2)源码:
import wordcloud #导入库模块
from PIL import Image
from imageio import imread
with open(r'C:\Users\Administrator\Desktop\202209408雷孟晓\Stu_pack\wordcloud\万疆.txt',encoding='utf-8')as file:#读取词云文本文件
fr=file.read()
im=imread('../Stu_pack/wordcloud/Love_Star.PNG') #读取遮罩图片
#im=image.open('../Stu_pack/wordcloud/Love_Star.PNG') 不能有这种方式读取遮罩图片
wc=wordcloud.WordCloud(mask=im,font_path='../Stu_pack/wordcloud/simhei.ttf',background_color='#ff0000') #创建词云对象
wc.generate(fr) #加载词云文本
wc.to_file('../R&Q_pic/test.png') #输出词云图片(文件)
Image.open('../R&Q_pic/test.png')#读取二进制图片文件
输出:
(3)源码:
import wordcloud #(1)导入库模块
#from PIL import Image
from imageio import imread
import matplotlib.pyplot as plt #导入数据可视化库里面的基础绘图库pyplot
with open(r'C:\Users\Administrator\Desktop\202209408雷孟晓\Stu_pack\wordcloud\万疆.txt',encoding='utf-8')as file:#(2)读取词云文本文件
fr=file.read()
im=imread('../Stu_pack/wordcloud/Love_Star.PNG') #读取遮罩图片
#im=image.open('../Stu_pack/wordcloud/Love_Star.PNG') 不能有这种方式读取遮罩图片
wc=wordcloud.WordCloud(mask=im,font_path='../Stu_pack/wordcloud/simhei.ttf',background_color='green') #(3)创建词云对象
wc.generate(fr) #(4)加载词云文本
plt.imshow(wc) #(5)绘制词语图片并显示在当前二维坐标系中
plt.axis('off') #隐藏坐标轴
plt.savefig('../R&Q_pic/test2.png')#将绘制图片文件保存到指定位置
plt.show() #显示绘制的图片
# wc.to_file('../R&Q_pic/test.png') #输出词云图片(文件)
# Image.open('../R&Q_pic/test.png')#读取二进制图片文件
输出: