python根据提供的电影评论数据集将txt格式的数据转换为csv格式的数据集
时间: 2023-09-18 16:08:03 浏览: 105
可以使用Python中的pandas库来将txt格式的电影评论数据集转换为csv格式的数据集。
首先,需要导入pandas库:
```python
import pandas as pd
```
然后,读取txt文件并将其转换为DataFrame对象:
```python
with open('movie_reviews.txt', 'r') as f:
lines = f.readlines()
data = {'review': [], 'label': []}
for line in lines:
review, label = line.strip().split('\t')
data['review'].append(review)
data['label'].append(label)
df = pd.DataFrame(data)
```
接着,将DataFrame对象保存为csv文件:
```python
df.to_csv('movie_reviews.csv', index=False)
```
完整代码如下:
```python
import pandas as pd
with open('movie_reviews.txt', 'r') as f:
lines = f.readlines()
data = {'review': [], 'label': []}
for line in lines:
review, label = line.strip().split('\t')
data['review'].append(review)
data['label'].append(label)
df = pd.DataFrame(data)
df.to_csv('movie_reviews.csv', index=False)
```
运行完毕后,会在当前目录下生成一个名为"movie_reviews.csv"的文件,即为转换后的csv格式的数据集。
阅读全文