python - 使用 Pandas 从 CSV 文件中提取值

Question

我有一个 CSV 文件，其中数据被组织成两列。

如何将负面情绪和正面情绪保存在单独的 csv 文件中。我的代码如下：

import pandas as pd 

df = pd.read_csv('~/Downloads/epinions3.csv')

df_neg = df.loc[['Neg']] # get all negative sentiments and then save to a file 

df_pos = # get all positive sentiments and save to a file 

print(df_neg)

标签： pythonpandascsv

Answer 1

df.loc[df['class'] == 'Neg'].to_csv('negative.csv', index=False)