将label2.csv文件中的两列提取出来存放到另一个csv文件中
时间: 2024-02-12 08:04:32 浏览: 65
将一个表格的数据保存到另一个表格
好的,可以使用Python中的pandas库来实现这个功能。具体的代码如下:
```python
import pandas as pd
# 读取label2.csv文件
data = pd.read_csv('label2.csv')
# 提取需要的两列数据
new_data = data[['file_name', 'label']]
# 将数据存储到新的csv文件中
new_data.to_csv('new_label.csv', index=False)
```
这段代码首先使用`pd.read_csv()`函数读取了`label2.csv`文件中的数据,并将其存储在`data`变量中。然后,使用`data[['file_name', 'label']]`语句提取了需要的两列数据,并将其存储在`new_data`变量中。最后,使用`new_data.to_csv()`函数将数据存储到名为`new_label.csv`的新文件中。`index=False`参数表示不需要保存行索引。
阅读全文