读取一个文件夹下所有csv文件中指定列名的信息,并且写入到一个excel中
时间: 2023-05-29 21:03:51 浏览: 189
以下是Python代码实现:
```python
import os
import pandas as pd
def read_csv_files(folder_path, column_names):
"""
读取一个文件夹下所有csv文件中指定列名的信息
:param folder_path: 文件夹路径
:param column_names: 列名列表
:return: 所有csv文件中指定列的信息列表
"""
data_list = []
for filename in os.listdir(folder_path):
if filename.endswith(".csv"):
file_path = os.path.join(folder_path, filename)
df = pd.read_csv(file_path)
data_list.append(df[column_names])
return pd.concat(data_list)
def write_to_excel(data, sheet_name, excel_path):
"""
将数据写入到excel中
:param data: 数据
:param sheet_name: 工作表名称
:param excel_path: excel文件路径
:return: None
"""
writer = pd.ExcelWriter(excel_path)
data.to_excel(writer, sheet_name=sheet_name, index=False)
writer.save()
if __name__ == '__main__':
folder_path = "/path/to/csv/files/folder"
column_names = ["column1", "column2"]
sheet_name = "Sheet1"
excel_path = "/path/to/output/excel/file.xlsx"
data = read_csv_files(folder_path, column_names)
write_to_excel(data, sheet_name, excel_path)
```
其中,`read_csv_files`函数用于读取所有csv文件中指定列名的信息,并将它们拼接到一个DataFrame中返回。`write_to_excel`函数用于将数据写入到excel文件中。
在主函数中,需要设置文件夹路径、列名列表、工作表名称和输出excel文件路径。然后,调用`read_csv_files`函数读取数据,再将数据写入到excel文件中。
阅读全文