请编写一段从某一个指定路径下的多个csv文件中提取第二列并将他们放入一个新的路径下的新的csv文件中的代码
时间: 2024-01-21 13:17:09 浏览: 114
源代码.txt
以下是Python代码示例,可以实现从指定路径下的多个CSV文件中提取第二列,并将它们放入一个新的路径下的新的CSV文件中:
```python
import os
import csv
# 指定原始CSV文件所在的路径
input_path = '/path/to/input/folder'
# 指定新CSV文件所在的路径
output_path = '/path/to/output/folder'
# 获取原始CSV文件路径下的所有文件名
files = os.listdir(input_path)
# 遍历每个CSV文件
for file in files:
if file.endswith('.csv'):
# 构造原始CSV文件的完整路径
input_file = os.path.join(input_path, file)
# 构造新CSV文件的完整路径
output_file = os.path.join(output_path, file)
# 读取原始CSV文件并提取第二列
with open(input_file, 'r') as csv_file:
reader = csv.reader(csv_file)
data = [row[1] for row in reader]
# 将提取的第二列写入新的CSV文件
with open(output_file, 'w') as csv_file:
writer = csv.writer(csv_file)
writer.writerow(data)
```
这段代码会遍历指定路径下的所有CSV文件,提取它们的第二列,并将所有提取出来的第二列写入一个新的CSV文件中,该文件位于指定的输出路径下。请注意,这段代码未考虑CSV文件中是否存在表头行,如果存在表头行,需要对代码进行适当修改。
阅读全文