读取和写入Excel

Halcom

环境：win7系统、64bit、CPU、Anaconda3-4.2.0-Windows-x86_64
软件：Anaconda3-4.2.0-Windows-x86_64、VS2013

读取和写入Excel：
xlrd和xlwt，习惯用array来操作

1. # -*- coding: utf-8 -*-
   
2. """
   
3. Created on Wed Jul 19 19:43:01 2017
   
4. 
   
5. @author: Administrator
   
6. """
   
7. 
   
8. import xlrd
   
9. import numpy
   
10. from matplotlib import pyplot as plt
    
11. from mpl_toolkits.mplot3d import Axes3D
    
12. excel = xlrd.open_workbook('ysw.xlsx');
    
13. # 获取第一个sheet
    
14. sheet1 = excel.sheets()[0]
    
15. 
    
16. # 获取数据行数
    
17. nrows = sheet1.nrows;
    
18. # 获取数据列数
    
19. ncols = sheet1.ncols;
    
20. 
    
21. # 获取第3列的数据，数据的列数从0开始计数
    
22. value = sheet1.col_values(2)
    
23. # 获取第3行的数据，数据的行列数从0开始计数
    
24. value1 = sheet1.row_values(2)
    
25. 
    
26. # 获取制定行列的数据
    
27. row = 2;
    
28. col = 2;
    
29. value2 = sheet1.cell(row,col).value;
    
30. 
    
31. # 获取整个excel表格数据
    
32. data = [[0 for x in range(ncols)]]*(nrows-1)
    
33. data = numpy.array(data,dtype='float64');
    
34. for i in range(nrows-1):
    
35.     data[i][0] = sheet1.cell(i+1,0).value;
    
36.     data[i][1] = sheet1.cell(i+1,1).value;
    
37.     data[i][2] = sheet1.cell(i+1,2).value;
    
38.     data[i][3] = sheet1.cell(i+1,3).value;
    
39.     data[i][4] = sheet1.cell(i+1,4).value;
    
40. 
    
41. # 转置
    
42. x,y,z = data.T[0], data.T[1],data.T[2]
    
43. fig = plt.figure();
    
44. # 创建一个三维的视图
    
45. ax = plt.subplot(111,projection='3d');
    
46. # 散点图
    
47. ax.scatter(x,y,z,c='b');
    
48. ax.set_xlabel("x")
    
49. ax.set_ylabel("y")
    
50. ax.set_zlabel("z")
    
51. ax.set_title("scatter diagram")
    
52. plt.show()
    
53. 
    
54. 
    
55. 
    
56. # 将data写入Excel
    
57. # 创建一个book
    
58. import xlwt
    
59. book = xlwt.Workbook(encoding='utf-8', style_compression=0);
    
60. # 在book里面添加一个sheet
    
61. sheet = book.add_sheet('halcom.cn',cell_overwrite_ok = True);
    
62. 
    
63. # 写入数据
    
64. sheet.write(0,0,'www.halcom.cn');
    
65. 
    
66. for i in range(12):
    
67.     sheet.write(i+1,0,data[i][0]);
    
68.     sheet.write(i+1,1,data[i][1]);         
    
69.     sheet.write(i+1,2,data[i][2]);
    
70.     sheet.write(i+1,3,data[i][3]);
    
71.     sheet.write(i+1,4,data[i][4]);
    
72.                
    
73. book.save('D:\Deep_learning_TensorFlow_python\Learning_DayByDay\Excel\data.xls')

复制代码

book.save需要写为.xls后缀的Excel。