从Excel工作簿中的表格动态构建Python列表

时间:2013-08-28 22:45:23

标签: python excel list

我正在尝试压缩我之前在python中编写的一些代码。我有一些抽出的代码循环遍历excel工作簿中的许多查找表。大约有20张工作簿中包含查找表。我想遍历每个查找表中的值并将它们添加到自己的列表中。我现有的代码如下所示:

test1TableList = []
for row in arcpy.SearchCursor(r"Z:\Excel\LOOKUP_TABLES.xlsx\LookupTable1$"):
    test1TableList.append(row.Code)

test2TableList = []
for row in arcpy.SearchCursor(r"Z:\Excel\LOOKUP_TABLES.xlsx\LookupTable1$"):
    test2TableList.append(row.Code)

test3TableList = []
for row in arcpy.SearchCursor(r"Z:\Excel\LOOKUP_TABLES.xlsx\LookupTable1$"):
    test3TableList.append(row.Code)

test4TableList = []
for row in arcpy.SearchCursor(r"Z:\Excel\LOOKUP_TABLES.xlsx\LookupTable1$"):
    test4TableList.append(row.Code)

test5TableList = []
for row in arcpy.SearchCursor(r"Z:\Excel\LOOKUP_TABLES.xlsx\LookupTable1$"):
    test5TableList.append(row.Code)

yadda yadda

我想压缩该代码(可能在函数中)。

要解决的问题:

  1. 工作表名称都不同。我需要循环遍历excel工作簿中的每个工作表,以便a)获取工作表对象和b)使用工作表名称作为python列表变量名称的一部分
  2. 我希望每个列表都保留在内存中,以便在代码中进一步使用
  3. 我一直在尝试这样的东西,但有效但python列表变量似乎不会留在内存中:

    import arcpy, openpyxl
    from openpyxl import load_workbook, Workbook
    
    wb = load_workbook(r"Z:\Excel\LOOKUP_TABLES.xlsx")
    
    for i in wb.worksheets:
    
        filepath = r"Z:\Excel\LOOKUP_TABLES.xlsx" + "\\" + i.title + "$"
        varList = []
        with arcpy.da.SearchCursor(filepath, '*') as cursor:
            for row in cursor:
                varList.append(row[0])
    
        # This is the area I am struggling with. I can't seem to find a way to return 
        # each list into memory. I've tried the following code to dynamically create 
        # variable names from the name of the sheet so that each list has it's own 
        # variable. After the code has run, I'd just like to set a print statement 
        # (i.e. print variablename1) which will return the list contained in the variable
        newList = str(i.title) + "List"
        newList2 = list(varList)
        print newList + " = " + str(newList2)
    

    我已经在这方面工作了一段时间,我毫不怀疑,在这一点上,我在思考我的解决方案,但我处于一个障碍。欢迎任何建议!

2 个答案:

答案 0 :(得分:3)

不确定它是否适合您,但您可以使用 pandas 将图纸导入数据框。

from pandas.io.excel import ExcelFile
filename = 'linreg.xlsx'
xl = ExcelFile(filename)

for sheet in xl.sheet_names:
    df = xl.parse(sheet)
    print df

答案 1 :(得分:1)

不使用育种清单,而是使用字典收集每张纸的数据:

import arcpy
from openpyxl import load_workbook

wb = load_workbook(r"Z:\Excel\LOOKUP_TABLES.xlsx")

sheets = {}
for i in wb.worksheets:
    filepath = r"Z:\Excel\LOOKUP_TABLES.xlsx" + "\\" + i.title + "$"
    with arcpy.da.SearchCursor(filepath, '*') as cursor:
        sheets[i.title] = [row[0] for row in cursor]

print sheets