1. 程式人生 > >Python 操作Excel 學習筆記(一)

Python 操作Excel 學習筆記(一)

#!C:/Python27
#coding=utf-8
import xlrd
import xlsxwriter


class readExcelFile(object):
    def __init__(self):
        #path = "D:/20170920-1.xls"
        print "初始化 >>> start ..."
    #開啟檔案讀取內容,返回結果
    def openExcelFile(self):
        path = r"D:/20170920-1.xls"
        try:
            data = xlrd.open_workbook(path)
            table = data.sheets()[0]
            print data.sheet_names()#列印sheet頁名稱
            for a in data.sheet_names():
                print a
            nrows = table.nrows  # 行數
            ncols = table.ncols  # 列數
            print nrows,ncols
            colnames = table.row_values(3)  # 某一行資料
            list = []
            for rownum in range(1, nrows):
                row = table.row_values(rownum)
               # print row
                for b in row:
                    #將每行資料整行列印
                    print b,
                #把每行資料放入list中打印出來
        except Exception, e:
            print str(e)
    def newExcelFile(self):
        path = 'demo.xlsx'
        # 建立excel檔案
        workbook = xlsxwriter.Workbook(path)
        #     新增worksheet,也可以指定名字
        worksheet = workbook.add_worksheet()
        worksheet.name = u'列表樣式繪圖'#修改第一列的名稱
        worksheet = workbook.add_worksheet('Test')
        # 設定第一列的寬度
        worksheet.set_column('A:A', len('hello ') + 1)
        # 新增一個加粗格式方便後面使用
        bold = workbook.add_format({'bold': True})
        # 在A1單元格寫入純文字
        worksheet.write('A1', 'Hello')
        # 在A2單元格寫入帶格式的文字
        worksheet.write('A2', 'World', bold)
        # 指定行列寫入數字,下標從0開始
        worksheet.write(2, 0, 123)
        worksheet.write(3, 0, 123.456)
        # 在B5單元格插入圖片
        worksheet.insert_image('B5', '165140.gif')
        workbook.close()
    def charts(self):
        workbook = xlsxwriter.Workbook('chart_column.xlsx')
        worksheet = workbook.add_worksheet()
        bold = workbook.add_format({'bold': 1})
        # 這是個資料table的列
        headings = ['Number', 'Batch 1', 'Batch 2']
        data = [
            [2, 3, 4, 5, 6, 7],
            [10, 40, 50, 20, 10, 50],
            [30, 60, 70, 50, 40, 30],
        ]
        # 寫入一行
        worksheet.write_row('A1', headings, bold)
        # 寫入一列
        worksheet.write_column('A2', data[0])
        worksheet.write_column('B2', data[1])
        worksheet.write_column('C2', data[2])
        ############################################
        # 建立一個圖表,型別是column
        chart1 = workbook.add_chart({'type': 'column'})
        # 配置series,這個和前面worksheet是有關係的。
        #     指定圖表的資料範圍
        chart1.add_series({
            'name': '=Sheet1!$B$1',
            'categories': '=Sheet1!$A$2:$A$7',
            'values': '=Sheet1!$B$2:$B$7',
        })
        chart1.add_series({
            'name': "=Sheet1!$C$1",
            'categories': '=Sheet1!$A$2:$A$7',
            'values': '=Sheet1!$C$2:$C$7',
        })
        #    配置series的另一種方法
        #     #     [sheetname, first_row, first_col, last_row, last_col]
        #     chart1.add_series({
        #         'name':         ['Sheet1',0,1],
        #         'categories':   ['Sheet1',1,0,6,0],
        #         'values':       ['Sheet1',1,1,6,1],
        #                        })
        #
        #
        #
        #     chart1.add_series({
        #         'name':       ['Sheet1', 0, 2],
        #         'categories': ['Sheet1', 1, 0, 6, 0],
        #         'values':     ['Sheet1', 1, 2, 6, 2],
        #     })
        #      新增圖表標題和標籤
        chart1.set_title({'name': 'Results of sample analysis'})
        chart1.set_x_axis({'name': 'Test number'})
        chart1.set_y_axis({'name': 'Sample length (mm)'})
        # 設定圖表風格
        chart1.set_style(11)
        # 在D2單元格插入圖表(帶偏移)
        worksheet.insert_chart('D2', chart1, {'x_offset': 25, 'y_offset': 10})
        #######################################################################
        #
        # 建立一個疊圖子型別
        chart2 = workbook.add_chart({'type': 'column', 'subtype': 'stacked'})
        # Configure the first series.
        chart2.add_series({
            'name': '=Sheet1!$B$1',
            'categories': '=Sheet1!$A$2:$A$7',
            'values': '=Sheet1!$B$2:$B$7',
        })
        # Configure second series.
        chart2.add_series({
            'name': '=Sheet1!$C$1',
            'categories': '=Sheet1!$A$2:$A$7',
            'values': '=Sheet1!$C$2:$C$7',
        })
        # Add a chart title and some axis labels.
        chart2.set_title({'name': 'Stacked Chart'})
        chart2.set_x_axis({'name': 'Test number'})
        chart2.set_y_axis({'name': 'Sample length (mm)'})
        # Set an Excel chart style.
        chart2.set_style(12)
        # Insert the chart into the worksheet (with an offset).
        worksheet.insert_chart('D18', chart2, {'x_offset': 25, 'y_offset': 10})

        #######################################################################
        #
        # Create a percentage stacked chart sub-type.
        #
        chart3 = workbook.add_chart({'type': 'column', 'subtype': 'percent_stacked'})
        # Configure the first series.
        chart3.add_series({
            'name': '=Sheet1!$B$1',
            'categories': '=Sheet1!$A$2:$A$7',
            'values': '=Sheet1!$B$2:$B$7',
        })
        # Configure second series.
        chart3.add_series({
            'name': '=Sheet1!$C$1',
            'categories': '=Sheet1!$A$2:$A$7',
            'values': '=Sheet1!$C$2:$C$7',
        })

        # Add a chart title and some axis labels.
        chart3.set_title({'name': 'Percent Stacked Chart'})
        chart3.set_x_axis({'name': 'Test number'})
        chart3.set_y_axis({'name': 'Sample length (mm)'})
        # Set an Excel chart style.
        chart3.set_style(13)
        # Insert the chart into the worksheet (with an offset).
        worksheet.insert_chart('D34', chart3, {'x_offset': 25, 'y_offset': 10})
        # 生成圓餅圖
        chart4 = workbook.add_chart({'type': 'pie'})
        # 定義資料
        data = [
            ['Pass', 'Fail', 'Warn', 'NT'],
            [333, 11, 12, 22],
        ]
        # 寫入資料
        worksheet.write_row('A51', data[0], bold)
        worksheet.write_row('A52', data[1])

        chart4.add_series({
            'name': u'介面測試報表圖',
            'categories': '=Sheet1!$A$51:$D$51',
            'values': '=Sheet1!$A$52:$D$52',
            'points': [
                {'fill': {'color': '#00CD00'}},
                {'fill': {'color': 'red'}},
                {'fill': {'color': 'yellow'}},
                {'fill': {'color': 'gray'}},
            ],
        })
        # Add a chart title and some axis labels.
        chart4.set_title({'name': u'介面測試統計'})
        chart4.set_style(3)
        #     chart3.set_y_axis({'name': 'Sample length (mm)'})
        try:
            worksheet.insert_chart('E52', chart4, {'x_offset': 25, 'y_offset': 10})
        except IOError as e:
            print e
        workbook.close()

if (__name__ == "__main__"):
    r = readExcelFile()
    #r.openExcelFile()
    #r.newExcelFile()
    r.charts()
    print('finished...')
    pass