读取文件夹中csv文件的第三列并保存
时间: 2024-01-13 08:03:39 浏览: 75
可以使用Python中的`csv`和`os`模块来读取文件夹中的CSV文件。以下是一个示例代码,可以读取文件夹中所有CSV文件的第三列并保存到一个新的CSV文件中:
```python
import csv
import os
directory = "csv_files" # CSV文件所在的文件夹
output_file = "output.csv" # 保存输出的CSV文件名
with open(output_file, "w", newline='') as out_csv_file: # 创建输出CSV文件
writer = csv.writer(out_csv_file)
for filename in os.listdir(directory): # 遍历文件夹中的所有文件
if filename.endswith(".csv"): # 如果是CSV文件
with open(os.path.join(directory, filename), newline='') as in_csv_file: # 打开CSV文件
reader = csv.reader(in_csv_file)
for row in reader:
if len(row) > 2: # 如果该行有至少3列
writer.writerow([row[2]]) # 写入第三列到输出CSV文件中
```
代码中,`directory`变量指定了CSV文件所在的文件夹,`output_file`变量指定了输出的CSV文件名。代码通过遍历文件夹中的所有文件,找到所有以`.csv`结尾的文件,打开文件并读取每一行的第三列,将其写入到输出CSV文件中。最终输出的CSV文件只包含第三列。
阅读全文