加载带有跳过情绪的 Pandas 数据框

Question

我有这个用于情感分析的数据集，使用以下代码加载数据：

url = 'https://raw.githubusercontent.com/jdvelasq/datalabs/master/datasets/amazon_cells_labelled.tsv'
df = pd.read_csv(url, sep='\t', names=["Sentence", "Feeling"])

问题是 DataFrame 与 NaN 匹配，但它只是整个句子的一部分。

现在的输出是这样的：

sentence                      feeling
I do not like it.             NaN
I give it a bad score.        0

输出应如下所示：

sentence                                    feeling
I do not like it. I give it a bad score     0

你能帮我根据分数连接或加载数据集吗？

Answer 1

在groupby 和agg 行之前创建虚拟组：

grp = df['Feeling'].notna().cumsum().shift(fill_value=0)
out = df.groupby(grp).agg({'Sentence': ' '.join, 'Feeling': 'last'})
print(out)

# Output:
                                                  Sentence  Feeling
Feeling                                                            
0        I try not to adjust the volume setting to avoi...      0.0
1                              Good case, Excellent value.      1.0
2        I thought Motorola made reliable products!. Ba...      1.0
3        When I got this item it was larger than I thou...      0.0
4                                        The mic is great.      1.0
...                                                    ...      ...
996      But, it was cheap so not worth the expense or ...      0.0
997      Unfortunately, I needed them soon so i had to ...      0.0
998      The only thing that disappoint me is the infra...      0.0
999      No money back on this one. You can not answer ...      0.0
1000     It's rugged. Well this one is perfect, at the ...      NaN

[1001 rows x 2 columns]