Bag of word scikit learn

import numpy as np
import pandas as pd
from sklearn.feature_extraction.text import CountVectorizer

docs = ['Tea is an aromatic beverage..',
        'After water, it is the most widely consumed drink in the world',
        'There are many different types of tea.',
        'Tea has a stimulating effect in humans.',
        'Tea originated in Southwest China during the Shang dynasty'] 

df = pd.DataFrame({'sms_message': docs, 'label': np.random.choice([0, 1], size=5)})

cv = CountVectorizer()
counts = cv.fit_transform(df['sms_message'])

df_counts = pd.DataFrame(counts.A, columns=cv.get_feature_names())
df_counts['label'] = df['label']

Source: stackoverflow.com

Similar codes

0 vote190 views1 answer
Python legend outside
Python
0 vote164 views1 answer
Change size of yticks python
Python
1 vote223 views1 answer
Python letter arr
Python
0 vote169 views1 answer
\n appears in json dump
Python
0 vote182 views1 answer
Python string: .strip()
Python
0 vote177 views1 answer
Mudopy
Python
0 vote174 views1 answer
Star program in python using for loop
Python
11 votes171 views3 answers
How to detect the reaction to a message discord.py
Python
0 vote180 views1 answer
Make y axis start at 0 python
Python
0 vote227 views1 answer
Numpy find columns containing nan
Python