使用python编程语言读取一个csv表格,然后分别将两个列表中的数据写入到这个csv表格中指定的两列,并保存该表格。指定的两列的列名分别是label和sentence_pair/target,要写入的列表分别是Label和target。
时间: 2024-04-12 10:27:13 浏览: 29
您可以使用Python的`csv`模块来读取和写入CSV文件。下面是一个示例代码,展示了如何读取CSV文件并将数据写入指定的两列。
```python
import csv
def write_to_csv(file_path, label_list, target_list):
with open(file_path, 'w', newline='') as csvfile:
fieldnames = ['label', 'sentence_pair/target']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()
for label, target in zip(label_list, target_list):
writer.writerow({'label': label, 'sentence_pair/target': target})
# 读取CSV文件
data = []
with open('input.csv', 'r') as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
data.append(row)
# 获取label和target列表
label_list = [row['Label'] for row in data]
target_list = [row['target'] for row in data]
# 写入CSV文件
write_to_csv('output.csv', label_list, target_list)
```
您只需要将`input.csv`替换为您要读取的CSV文件的路径,将`output.csv`替换为您要保存的CSV文件的路径。`label_list`和`target_list`是您要写入CSV文件的两个列表。
请注意,这个示例假设CSV文件的第一行是列名。如果不是,请根据实际情况修改`write_to_csv`函数中的`writeheader()`行。