帮我用python生成一个程序,要求读取一个文件夹中的所有csv文件,并将每个文件的第6列都提取出来,组成一个新的csv文件
时间: 2024-11-05 13:26:06 浏览: 30
在Python中,我们可以使用`os`模块来遍历文件夹,`pandas`库来处理CSV文件。下面是一个简单的示例程序,它会读取指定文件夹下的所有`.csv`文件,提取每个文件的第6列,并将结果写入新的csv文件:
```python
import os
import pandas as pd
def extract_column_from_csvs(input_folder, output_file):
# 创建一个空列表存储所有文件的第6列
all_data = []
# 遍历输入文件夹中的所有CSV文件
for filename in os.listdir(input_folder):
if filename.endswith('.csv'):
filepath = os.path.join(input_folder, filename)
# 读取CSV文件并获取第6列(注意,从0开始计数)
df_temp = pd.read_csv(filepath, usecols=[5])
# 将第6列添加到总数据列表中
all_data.append(df_temp.iloc[:, 0])
# 合并所有数据为一个大DataFrame
combined_df = pd.concat(all_data)
# 写入新csv文件
combined_df.to_csv(output_file, index=False)
# 调用函数,传入你的输入文件夹路径和新csv的输出路径
extract_column_from_csvs('input_folder_path', 'output.csv')
阅读全文
相关推荐
![zip](https://img-home.csdnimg.cn/images/20241231045053.png)
![pdf](https://img-home.csdnimg.cn/images/20241231044930.png)
![pdf](https://img-home.csdnimg.cn/images/20241231044930.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![-](https://img-home.csdnimg.cn/images/20241226111658.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)