文件夹中有九百多个csv文件,写出如何做到一个csv文件对应一个标签的代码
时间: 2024-05-03 10:17:50 浏览: 107
假设每个csv文件的文件名即为对应的标签,可以使用Python的os模块和pandas库来实现。具体代码如下:
```python
import os
import pandas as pd
# 获取文件夹中所有csv文件的文件名
folder_path = '/path/to/folder'
files = os.listdir(folder_path)
csv_files = [f for f in files if f.endswith('.csv')]
# 遍历每个csv文件,读取数据并添加标签列
for csv_file in csv_files:
label = os.path.splitext(csv_file)[0] # 获取文件名作为标签
file_path = os.path.join(folder_path, csv_file)
df = pd.read_csv(file_path)
df['label'] = label # 添加标签列
# 保存新的csv文件,文件名为原文件名加上_label后缀
new_file_name = os.path.splitext(csv_file)[0] + '_label.csv'
new_file_path = os.path.join(folder_path, new_file_name)
df.to_csv(new_file_path, index=False)
```
运行以上代码后,每个原始的csv文件都会生成一个新的csv文件,新文件名为原文件名加上_label后缀,其中包含了添加了标签列的数据。这样每个csv文件就对应一个标签了。
阅读全文