同一文件夹下批量提取csv数据中的中的某三列,并保存到新的文件
时间: 2023-06-27 11:06:45 浏览: 102
批量读取csv文件 并保存成mat格式
可以使用Python的pandas库来处理csv文件。以下是一个示例代码,可以批量提取csv数据中的某三列并保存到新的文件。
```python
import pandas as pd
import os
# 需要提取的列的列名
cols_to_keep = ['col1', 'col2', 'col3']
# 待处理的csv文件所在的文件夹路径
input_folder = '/path/to/folder'
# 新文件保存的文件夹路径
output_folder = '/path/to/output/folder'
# 获取文件夹中所有csv文件的文件名
csv_files = [f for f in os.listdir(input_folder) if f.endswith('.csv')]
# 循环处理每个csv文件
for csv_file in csv_files:
# 读取csv文件
df = pd.read_csv(os.path.join(input_folder, csv_file))
# 只保留需要的列
df = df[cols_to_keep]
# 保存到新的文件
output_file = os.path.join(output_folder, csv_file)
df.to_csv(output_file, index=False)
```
这段代码会读取`/path/to/folder`文件夹下所有以`.csv`结尾的文件,提取`col1`、`col2`和`col3`三列,然后保存到`/path/to/output/folder`文件夹下,文件名与原文件相同。请根据实际情况修改代码中的文件夹路径和列名。
阅读全文