在只读模式下使用OpenPyXL获取Excel工作表的列名

时间:2018-08-22 22:17:32

标签: python excel openpyxl

我怎么找

  1. openpyxl Read-only 工作表中的列名(第一行单元格的值)?
      在以下示例工作表中
    • CityPopulationCountry
  2. openpyxl只读工作簿中的所有列名称?
    • CityPopulationCountry,工作表1中的框架和所有其他工作表中的其他列名

Excel工作表示例:

| City       | Population  |    Country   |
| -----------|------------ | ------------ |
| Madison    |   252,551   |     USA      |
| Bengaluru  | 10,178,000  |    India     |
| ...        |       ...   |     ...      |

示例代码:

from openpyxl import load_workbook

wb = load_workbook(filename=large_file.xlsx, read_only=True)
sheet = wb.worksheets[0]

... (not sure where to go from here)

注意:

  • 我必须使用只读,因为Excel文件具有超过一百万行(不要问)
  • 我想要列名称,以便最终推断出列类型并将excel数据导入PostgreSQL数据库

4 个答案:

答案 0 :(得分:5)

这将打印第1行中的所有内容;

list_with_values=[]
for cell in ws[1]:
    list_with_values.append(cell.value)

如果出于某种原因想要获得要填写的列字母的列表,则可以:

column_list = [cell.column for cell in ws[1]]

关于第二个问题; 假设您已将标头值存储在名为“ list_with_values”的列表中

from openpyxl import Workbook
wb = Workbook()
ws = wb['Sheet']
#Sheet is the default sheet name, you can rename it or create additional ones with wb.create_sheet()
ws.append(list_with_values)
wb.save('OutPut.xlsx')

答案 1 :(得分:2)

只读模式提供对工作表中任何一行或一组行的快速访问。使用方法iter_rows()限制选择。因此,要获得工作表的第一行:

rows = ws.iter_rows(min_row=1, max_row=1) # returns a generator of rows
first_row = next(rows) # get the first row
headings = [c.value for c in first_row] # extract the values from the cells

答案 2 :(得分:1)

Charlie Clarks 的答案压缩成一个带有列表理解的单行

    headers = [c.value for c in next(wb['sheet_name'].iter_rows(min_row=1, max_row=1))]

答案 3 :(得分:0)

我就是这样处理的

from openpyxl.utils import get_column_letter

def get_columns_from_worksheet(ws):
  return {
      cell.value: {
          'letter': get_column_letter(cell.column),
          'number': cell.column - 1
      } for cell in ws[1] if cell.value
  }

使用的一个例子是

from openpyxl import load_workbook

wb = load_workbook(filename='my_file.xlsx')
ws = wb['MySheet']

COLUMNS = get_columns_from_worksheet(ws)

for cell in ws[COLUMNS['MY Named Column']['letter']]:
    print(cell.value)

同时捕获字母和数字代码的主要原因是因为 openpyxl 中不同的函数和模式使用数字或字母,因此参考两者是无价的