python写入EXCEL单元格范围(NumPy?)

时间:2014-09-22 14:28:17

标签: python excel numpy

我正在尝试将使用NumPy编写的矩阵写入EXCEL文件。我需要指定必须写入矩阵的一系列单元格。

我需要将矩阵写入EXCEL文件第4页中的单元格A4:Z512。

现在,标准的EXCEL文件有3张,所以我需要先添加第4张,然后将矩阵写入其中。

有没有办法在python 2.7中执行此操作?是否可以使用纯NumPy进行此操作?

1 个答案:

答案 0 :(得分:1)

我没有使用过NumPy,所以我不确定你是否可以操作excel文件。但是对于处理excel文件,我建议使用win32com库。下面是我过去使用的一些代码,使win32com API更易于使用。随意使用自己的代码。希望这有帮助!

import win32com.client as win32

excel = win32.gencache.EnsureDispatch('Excel.Application')

def openExcel(makeExcelVisible=True):
    excel.Visible = makeExcelVisible

def closeExcel():
    excel.Application.Quit()

class ExcelFile(object):
# opens up a workbook to work on, not selecting a file name creates a new one
    def __init__(self, fileName=None):
        if fileName == None:
            self.wb = excel.Workbooks.Add()
        else:
            self.wb = excel.Workbooks.Open(fileName)
        self.ws = None

    def addWorksheet(self):
    # adds a new worksheet to the workbook and makes it the current worksheet
        self.ws = self.wb.Worksheets.Add()

    def selectWorksheet(self, worksheetName):
    # selects a worksheet to work on
        self.ws = self.wb.Worksheets(worksheetName)

    def renameWorksheet(self, worksheetName):
    # renames current worksheet
        self.ws.Name = worksheetName

    def save(self):
    # saves the workbook
        self.wb.Save()

    def saveAs(self, fileName):
    # saves the workbook with the file name
        self.wb.SaveAs(fileName)

    def close(self):
    # closes the workbook
        self.wb.Close()

    def insertIntoCell(self, cellRow, cellCol, data):
        self.ws.Cells(cellRow,cellCol).Value = data

    def clearCell(self, cellRow, cellCol):
        self.ws.Cells(cellRow,cellCol).Value = None

以下是如何使用上述代码的示例。它创建一个新的excel文件,重命名工作表,将信息添加到每个工作表的第一个单元格中,并将文件保存为“test.xlsx”。默认保存位置是您的主目录。

worksheets = ["Servers", "Printers", "Drives", "IT Phones"]
information = ["Server WS", "Printer WS", "Driver WS", "IT Phone WS"]

def changeCells(information):
    excelFile = ExcelFile()
    for index in xrange(len(worksheets)):
        sheetNumber = index + 1
        if sheetNumber == 4:
            excelFile.addWorksheet()
        excelFile.selectWorksheet("Sheet%d" % sheetNumber)
        excelFile.renameWorksheet(worksheets[index])
        excelFile.insertIntoCell(1,1,information[index])
    excelFile.saveAs("test.xlsx")
    excelFile.close()

openExcel()
changeCells(information)
closeExcel()

另外,我建议您自己查看win32com的API。这是一个非常好用的图书馆。

我将在Sheet4上输入矩阵所需的实际代码汇总到A4:Z512。

def addMatrix(matrix):
    # use ExcelFile("fileName.xlsx") if you need to add to a specific file
    excelFile = ExcelFile()
    excelFile.addWorksheet()
    # default excel files only have 3 sheets so had to add one
    excelFile.selectWorksheet("Sheet4")
    # xrange(4,513) since end index is exclusive
    for row in xrange(4,513):
        # 1 for A, 26 for Z
        for col in xrange(1,27):
            mRow = row - 4
            mCol = col - 1
            excelFile.insertIntoCell(row, col, matrix[mRow][mCol])
    excelFile.saveAs("test.xlsx")
    excelFile.close()

matrix = list()
for row in xrange(509):
    matrix.append([])
    for col in xrange(26):
        matrix[row].append(0)
# the matrix is now filled with zeros

# use openExcel(False) to run faster, it won't open a window while running
openExcel()
addMatrix(matrix)
closeExcel()