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

12 votes212 views1 answer
How to create an env to use in vs code using anaconda
Python
0 vote217 views1 answer
Matplotlib show grid for log or logit
Python
1 vote245 views1 answer
How to create file using python cat command
Python
0 vote204 views1 answer
Python numpy array subtract
Python
-1 vote206 views1 answer
How to write pretty xml to a file python
Python
6 votes209 views1 answer
Pandas change date format to yyyy-mm-dd
Python
0 vote216 views2 answers
Historical tick bid ask
Python
0 vote171 views1 answer
How to use constructor of base class in derived class in python
Python
0 vote203 views1 answer
How to change the size of datapoint in plot python
Python
0 vote205 views1 answer
How to break an if statement python
Python