Чтение файла Excel в Python
У меня есть файл Excel
Arm_id DSPName DSPCode HubCode PinCode PPTL
1 JaVAS 01 AGR 282001 1,2
2 JaVAS 01 AGR 282002 3,4
3 JaVAS 01 AGR 282003 5,6
Я хочу сохранить строку в форме Arm_id,DSPCode,Pincode
. Этот формат настраивается, т.е. Он может измениться на DSPCode,Arm_id,Pincode
. Я сохраняю его в списке, например
FORMAT = ['Arm_id', 'DSPName', 'Pincode']
Как я могу прочитать содержимое определенного столбца с предоставленным именем, учитывая, что FORMAT
настраивается.
Это то, что я пробовал. В настоящее время я могу читать все содержимое в файле
from xlrd import open_workbook
wb = open_workbook('sample.xls')
for s in wb.sheets():
#print 'Sheet:',s.name
values = []
for row in range(s.nrows):
col_value = []
for col in range(s.ncols):
value = (s.cell(row,col).value)
try : value = str(int(value))
except : pass
col_value.append(value)
values.append(col_value)
print values
Мой вывод
[[u'Arm_id', u'DSPName', u'DSPCode', u'HubCode', u'PinCode', u'PPTL'], ['1', u'JaVAS', '1', u'AGR', '282001', u'1,2'], ['2', u'JaVAS', '1', u'AGR', '282002', u'3,4'], ['3', u'JaVAS', '1', u'AGR', '282003', u'5,6']]
Затем я обхожусь вокруг values[0]
, пытаясь узнать содержимое FORMAT
в values[0]
, а затем получая индекс Arm_id, DSPname and Pincode
в values[0]
, а затем из следующего цикла. Я знаю индекс всех FORMAT
, тем самым узнав, какую ценность мне нужно получить.
Но это такое плохое решение.
Как получить значения определенного столбца с именем в файле excel?
Ответы
Ответ 1
Это один из подходов:
from xlrd import open_workbook
class Arm(object):
def __init__(self, id, dsp_name, dsp_code, hub_code, pin_code, pptl):
self.id = id
self.dsp_name = dsp_name
self.dsp_code = dsp_code
self.hub_code = hub_code
self.pin_code = pin_code
self.pptl = pptl
def __str__(self):
return("Arm object:\n"
" Arm_id = {0}\n"
" DSPName = {1}\n"
" DSPCode = {2}\n"
" HubCode = {3}\n"
" PinCode = {4} \n"
" PPTL = {5}"
.format(self.id, self.dsp_name, self.dsp_code,
self.hub_code, self.pin_code, self.pptl))
wb = open_workbook('sample.xls')
for sheet in wb.sheets():
number_of_rows = sheet.nrows
number_of_columns = sheet.ncols
items = []
rows = []
for row in range(1, number_of_rows):
values = []
for col in range(number_of_columns):
value = (sheet.cell(row,col).value)
try:
value = str(int(value))
except ValueError:
pass
finally:
values.append(value)
item = Arm(*values)
items.append(item)
for item in items:
print item
print("Accessing one single value (eg. DSPName): {0}".format(item.dsp_name))
print
Вам не нужно использовать пользовательский класс, вы можете просто взять dict()
. Однако, если вы используете класс, вы можете получить доступ ко всем значениям с помощью точечной нотации, как вы видите выше.
Вот вывод script выше:
Arm object:
Arm_id = 1
DSPName = JaVAS
DSPCode = 1
HubCode = AGR
PinCode = 282001
PPTL = 1
Accessing one single value (eg. DSPName): JaVAS
Arm object:
Arm_id = 2
DSPName = JaVAS
DSPCode = 1
HubCode = AGR
PinCode = 282002
PPTL = 3
Accessing one single value (eg. DSPName): JaVAS
Arm object:
Arm_id = 3
DSPName = JaVAS
DSPCode = 1
HubCode = AGR
PinCode = 282003
PPTL = 5
Accessing one single value (eg. DSPName): JaVAS
Ответ 2
Несколько поздний ответ, но с пандами можно получить непосредственно столбец файла Excel:
import pandas
import xlrd
df = pandas.read_excel('sample.xls')
#print the column names
print df.columns
#get the values for a given column
values = df['Arm_id'].values
#get a data frame with selected columns
FORMAT = ['Arm_id', 'DSPName', 'Pincode']
df_selected = df[FORMAT]
Ответ 3
Таким образом, ключевые части должны захватить заголовок (col_names = s.row(0)
) и при повторении через строки пропустить первую строку, которая не нужна for row in range(1, s.nrows)
- сделать, используя диапазон от 1 до (не неявный 0). Затем вы используете zip для перехода через строки, содержащие "имя" в качестве заголовка столбца.
from xlrd import open_workbook
wb = open_workbook('Book2.xls')
values = []
for s in wb.sheets():
#print 'Sheet:',s.name
for row in range(1, s.nrows):
col_names = s.row(0)
col_value = []
for name, col in zip(col_names, range(s.ncols)):
value = (s.cell(row,col).value)
try : value = str(int(value))
except : pass
col_value.append((name.value, value))
values.append(col_value)
print values
Ответ 4
Используя панд, мы можем легко читать Excel.
import pandas as pd
import xlrd as xl
from pandas import ExcelWriter
from pandas import ExcelFile
DataF=pd.read_excel("Test.xlsx",sheet_name='Sheet1')
print("Column headings:")
print(DataF.columns)
Тест на: https://repl.it Ссылка: https://pythonspot.com/read-excel-with-pandas/
Ответ 5
Подход, который я принял, считывает информацию заголовка из первой строки, чтобы определить индексы интересующих столбцов.
Вы упомянули в вопросе, что вы также хотите, чтобы значения выводились в строку. Я динамически строю строку формата для вывода из списка столбцов FORMAT. Строки добавляются к строке значений, разделенной новой строкой char.
Порядок выходных столбцов определяется порядком имен столбцов в списке FORMAT.
В моем коде ниже важно имя столбца в списке FORMAT. В вышеприведенном вопросе у вас есть "Pincode" в вашем списке FORMAT, но "PinCode" в вашем превосходстве. Это не будет работать ниже, это должно быть "PinCode".
from xlrd import open_workbook
wb = open_workbook('sample.xls')
FORMAT = ['Arm_id', 'DSPName', 'PinCode']
values = ""
for s in wb.sheets():
headerRow = s.row(0)
columnIndex = [x for y in FORMAT for x in range(len(headerRow)) if y == firstRow[x].value]
formatString = ("%s,"*len(columnIndex))[0:-1] + "\n"
for row in range(1,s.nrows):
currentRow = s.row(row)
currentRowValues = [currentRow[x].value for x in columnIndex]
values += formatString % tuple(currentRowValues)
print values
Для ввода образца, который вы указали выше, выводите этот код:
>>> 1.0,JaVAS,282001.0
2.0,JaVAS,282002.0
3.0,JaVAS,282003.0
И потому, что я python noob, реквизиты должны быть:
этот ответ,
этот ответ,
этот вопрос,
этот вопрос
и этот ответ.
Ответ 6
Хотя я почти всегда просто использую панды для этого, мой текущий маленький инструмент упакован в исполняемый файл, и включение панд является излишним. Поэтому я создал версию решения poida, в результате которой появился список именованных кортежей. Его код с этим изменением будет выглядеть так:
from xlrd import open_workbook
from collections import namedtuple
from pprint import pprint
wb = open_workbook('sample.xls')
FORMAT = ['Arm_id', 'DSPName', 'PinCode']
OneRow = namedtuple('OneRow', ' '.join(FORMAT))
all_rows = []
for s in wb.sheets():
headerRow = s.row(0)
columnIndex = [x for y in FORMAT for x in range(len(headerRow)) if y == headerRow[x].value]
for row in range(1,s.nrows):
currentRow = s.row(row)
currentRowValues = [currentRow[x].value for x in columnIndex]
all_rows.append(OneRow(*currentRowValues))
pprint(all_rows)
Ответ 7
Вот код для чтения файла Excel и печати всех ячеек, представленных в столбце 1 (кроме первой, то есть заголовка):
import xlrd
file_location="C:\pythonprog\xxx.xlsv"
workbook=xlrd.open_workbook(file_location)
sheet=workbook.sheet_by_index(0)
print(sheet.cell_value(0,0))
for row in range(1,sheet.nrows):
print(sheet.cell_value(row,0))