使用pandas中的read_excel方法将excel文件导入时,希望保留excel表格中的行和列名称。例如,我希望将我的列命名为“A”、“B”、...、“Z”、“AA”、“AB”等,将行命名为从1开始的数字。请问有什么好的方法可以实现这个功能吗?
你需要定制 映射 并将其应用到 rename
:
np.random.seed(100)
df = pd.DataFrame(np.random.randint(10, size=(5,5)))
print (df)
0 1 2 3 4
0 8 8 3 7 7
1 0 4 2 5 2
2 2 2 1 0 8
3 4 0 9 6 2
4 4 1 5 3 4
def colToExcel(col): # col is 1 based
excelCol = str()
div = col
while div:
(div, mod) = divmod(div-1, 26) # will return (x, 0 .. 25)
excelCol = chr(mod + 65) + excelCol
return excelCol
df = df.rename(index=lambda x: x+1, columns=lambda y: colToExcel(y+1) )
print (df)
A B C D E
1 8 8 3 7 7
2 0 4 2 5 2
3 2 2 1 0 8
4 4 0 9 6 2
5 4 1 5 3 4
替代方案:
def conv(num):
convStr = "ABCDEFGHIJKLMNOPQRSTUVWXYZ" # Assign any base you'd like
b = len(convStr)
if num<b:
return convStr[num]
else:
return conv(num//b-1) + convStr[num%b]
df = df.rename(index=lambda x: x+1, columns=lambda y: colToExcel(y) )
def converter(dict):
converter = {}
for key in dict:
converter[key] = str
return converter
columntype_table = {'A': datatype.str,
'B': datatype.str,
'C': datatype.str,
'D': datatype.str,
'E': datatype.str,
'F': datatype.str,
'G': datatype.str,
'H': datatype.str}
data = pandas.read_excel(fileX, encoding=X, sheetname=Y, converters=converterDic(columns_dict), header=None)
read_excel
。这是一个难题,也许不容易在pandas中实现。 - jezrael