使用 groupby 进行字数统计然后使用 python 循环的问题

Question

字数和累计和

我有一个多达 150 万行的数据集。该数据集是一个时间序列，是一个年份格式，如下所示。我正在尝试以累积格式计算每年的字符串。示例如下：

lodgement_year                trademark_text
  1906                          PEPS
  1906  BILE BEANS FOR BILIOUSNESS B
  1906                     ZAM-BUK Z

  lodgement_year
  1906    {PEPS BILE BEANS FOR BILIOUSNESS B ZAM-BUK Z Z...

  1907    {WHS CHERUB BLACK & WHITE SOUTHERN CROSS HISTO...

作为初始任务，我将字符串分组，然后使用 xxx 在此论坛中发布的代码全年应用循环。当循环工作时，会立即出现以下消息：

    The code :

    d = df_merge.groupby('lodgement_year')['trademark_text'].apply(lambda x: "{%s}" % ' '.join(x))


  for name in d.index:
            data = d.loc[name]
            ngram_vectorizer =    CountVectorizer(analyzer='word',tokenizer=word_tokenize, ngram_range=(1, 1), min_df=1)              
            X = ngram_vectorizer.fit_transform(data.split('\n'))
            vocab = list(ngram_vectorizer.get_feature_names())
            counts = X.sum(axis=0).A1
            freq_distribution = Counter(dict(zip(vocab, counts)))
            print (name, freq_distribution.most_common(10))

错误信息：

Traceback（最近一次调用最后一次）：

文件“/Users/PycharmProjects/Slice_Time_Series”，第 65 行，在 X = ngram_vectorizer.fit_transform(data.split('\n'))

文件“/Users/anaconda3/lib/python3.6/site-packages/pandas/core/generic.py”，第 3081 行，在 getattr 中 返回对象。getattribute（自我，姓名） AttributeError：“系列”对象没有属性“拆分”

错误前的输出：

 1906 [('.', 24), ("'s", 22), ('star', 18), ('&', 15), ('kodak', 12), ('co', 9), ('the', 9), ('brand', 8), ('express', 8), ('anchor', 6)]
 1907 [('&', 11), ("'s", 11), ('brand', 11), ('pinnacle', 7), ('vaseline', 7), ('the', 6), ('.', 5), ('co.', 5), ('kepler', 5), ('lucas', 5)]

任何帮助将不胜感激。作为下一个任务，我尝试创建一个系列，该系列是 1906 和 1906 加上 1907 然后 1906+1907+1908 的总和，我不知道该怎么做，任何指导也会很棒。

伊恩

Answer 1

您可以先通过 lodgement 变量对您的 DF 进行分组，然后遍历名称-组对，再进行一次 groupby，最后加入结果。

import pandas as pd
from collections import Counter
df = pd.DataFrame({'lodg':[1,2,1,3,1,2,2,3,1,1],
               'text':['x y z','y y','x y','z x','y x','y y z','x z','x x','x x','y z']})
grouped = df.groupby('lodg')
joined = []
for name, group in grouped:
    texts = " ".join(group['text'])
    groupCounts = Counter(texts.split(" ")).items()
    joined.append([name, [texts], groupCounts])
groupedJoined = pd.DataFrame(joined, columns = ['lodg','texts','textCounts'])

结果：

In [16]: groupedJoined
Out[16]: 
   lodg                    texts                textCounts
0     1  [x y z x y y x x x y z]  [(y, 4), (x, 5), (z, 2)]
1     2          [y y y y z x z]  [(y, 4), (x, 1), (z, 2)]
2     3                [z x x x]          [(x, 3), (z, 1)]

在显式解决方案之后，花了一点时间才想出 1-liner lambda 解决方案。

两个 Lambda：

df.groupby('lodg')['text'].apply(lambda x: "%s" % ' '.join(x)).apply(lambda x: Counter(x.split(" ")).items())

单个 Lambda：

df.groupby('lodg')['text'].apply(lambda x: Counter((' '.join(x)).split(" ")).items())

两者产生相同的结果：

Out[62]: 
lodg
1    [(y, 4), (x, 5), (z, 2)]
2    [(y, 4), (x, 1), (z, 2)]
3            [(x, 3), (z, 1)]
Name: text, dtype: object

现在，如果您不想处理未命名的列（实际上称为索引），请将结果命名为 res，然后执行 res.reset_index() 以获得此结果：

In [68]: res.reset_index()
Out[68]: 
   lodg                      text
0     1  [(y, 4), (x, 5), (z, 2)]
1     2  [(y, 4), (x, 1), (z, 2)]
2     3          [(x, 3), (z, 1)]

这应该更容易使用。