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 vote127 views1 answer
Cannot import name 'joblib' from 'sklearn.externals' (/shared-libs/python3.7/py/lib/python3.7/site-packages/sklearn/externals/__init__.py)
Python
3 votes167 views1 answer
Pandas concat and reset index
Python
0 vote212 views1 answer
Python conditions symbols
Python
0 vote173 views1 answer
How to display text on boxplot in python
Python
9 votes164 views7 answers
Print python
Python
13 votes131 views2 answers
Pd.set_option('display.max_columns' none)
Python
1 vote170 views1 answer
Python swap two variables
Python
29 votes194 views2 answers
Python initialize array
Python
0 vote177 views1 answer
Python how to change a point in a multidimensional list
Python
0 vote185 views1 answer
Fast guess for divisible numbers between two numbers
Python