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

1 vote297 views1 answer
Nlargest hierarchy series pandas
Python
4 votes280 views1 answer
Record the amount of time ittales for code to run python
Python
3 votes293 views2 answers
Find all indices of element in string python
Python
11 votes347 views6 answers
For enumerate python
Python
0 vote298 views1 answer
Spacy nlp load
Python
-2 votes275 views1 answer
How to create multiple dictionaries in python
Python
1 vote302 views1 answer
Python multiaxis slicing
Python
20 votes337 views6 answers
Python generate random password
Python
0 vote290 views1 answer
Python get next item from generator
Python
0 vote284 views1 answer
Discord.py APK
Python