我正在使用Python 3.4和xlrd。在处理数据之前,我希望按照主列对Excel表进行排序。是否有相关的库可以实现这个功能?
xlrd
。最大的缺点是它不能本地写入XLSX格式。xlrd
和来自此答案的一些修改:import xlwt
from xlrd import open_workbook
target_column = 0 # This example only has 1 column, and it is 0 indexed
book = open_workbook('test.xlsx')
sheet = book.sheets()[0]
data = [sheet.row_values(i) for i in xrange(sheet.nrows)]
labels = data[0] # Don't sort our headers
data = data[1:] # Data begins on the second row
data.sort(key=lambda x: x[target_column])
bk = xlwt.Workbook()
sheet = bk.add_sheet(sheet.name)
for idx, label in enumerate(labels):
sheet.write(0, idx, label)
for idx_r, row in enumerate(data):
for idx_c, value in enumerate(row):
sheet.write(idx_r+1, idx_c, value)
bk.save('result.xls') # Notice this is xls, not xlsx like the original file is
这将输出以下工作簿:
另一个选项(也可以利用XLSX输出)是使用pandas
。 代码也更简洁:
import pandas as pd
xl = pd.ExcelFile("test.xlsx")
df = xl.parse("Sheet1")
df = df.sort(columns="Header Row")
writer = pd.ExcelWriter('output.xlsx')
df.to_excel(writer,sheet_name='Sheet1',columns=["Header Row"],index=False)
writer.save()
这将输出:
在 to_excel
函数中,设置了 index
为 False
,因此 Pandas dataframe 的索引不会包含在 Excel 文档中。其余的关键字应该是很容易理解的。
我只是想更新一下答案,因为Pandas的实现方式随着时间的推移已经有所改变。以下是现在应该可以工作的代码(Pandas 1.1.2)。
import pandas as pd
xl = pd.ExcelFile("test.xlsx")
df = xl.parse("Sheet1")
df = df.sort_values(by="Header Row")
...
sort
函数现在称为sort_by
,而columns
被替换为by
。
sorted(list)
进行排序。 - runDOSrun