python 只保存excel指定列函数 具体例子
时间: 2023-07-21 19:06:38 浏览: 177
当你想要保存 Excel 表中的指定列时,可以使用 `pandas` 库。下面是一个具体的例子:
```python
import pandas as pd
def save_selected_columns(file_path, sheet_name, selected_columns, output_file):
# 读取 Excel 文件中的指定工作表
df = pd.read_excel(file_path, sheet_name=sheet_name)
# 选择指定的列
df_selected = df[selected_columns]
# 将选择的列保存到新的 Excel 文件中
df_selected.to_excel(output_file, index=False)
```
我们以一个名为 "data.xlsx" 的 Excel 文件为例,其中包含了名为 "Sheet1" 的工作表。假设我们希望只保存其中的 "Name" 和 "Age" 两列到新的 Excel 文件 "output.xlsx" 中,可以按照以下方式调用函数:
```python
save_selected_columns("data.xlsx", "Sheet1", ["Name", "Age"], "output.xlsx")
```
这样,函数将会读取 "data.xlsx" 文件中的 "Sheet1" 工作表,并且只保留 "Name" 和 "Age" 两列,最后将它们保存到名为 "output.xlsx" 的新文件中。请确保已经安装了 `pandas` 和 `openpyxl` 库。
阅读全文