python - 使用 pandas 读取和编辑 Excel 文件

Question

因此，我正在读取 xlsx 文件，需要选择 3 个特定列，然后重命名它们，并将每列的前 100 行传输到新的 xlsx 文件，并将它们从原始文件中删除。

到目前为止，我的代码正在从手动编辑的 xlsx 文件中读取，我在其中重命名列，选择其中的前 100 个(标题除外)等:

df = pandas.read_excel('test.xlsx', names=['date', 'value', 'source'])
df['date'] = pandas.to_datetime(df['date'], format='%b %d %Y.%f').astype(str)

print(df.head(5))

正如预期的那样返回:

         date    value               source

0  2018-05-01      1                xxxxxxx
1  2018-05-01      1      xxxxxxxxxxxxxxxxx
2  2018-05-02      1      xxxxxxxxxxxxxxxxx
3  2018-05-02      1      xxxxxxxxxxxxxxxxx
4  2018-05-03      1     xxxxxxxxxxxxxxxxxx

有没有办法用pandas自动读取和编辑xlsx文件？其他库？我是 python 新手，我真的迷失了:(

Answer 1

似乎您需要 2 个文件输出。一个包含前一百行，另一个包含剩余的行。这很简单 pandas.DataFrame.iloc :

import pandas as pd

df = pd.read_excel('test.xlsx', names=['date', 'value', 'source'])

df.iloc[:100].to_excel('out1.xlsx', index=False)
df.iloc[100:].to_excel('out2.xlsx', index=False)