How to create 2D array in python

Question

I'm trying to create a function called "words_in_texts" to get the result like this

words_in_texts(['hello', 'bye', 'world'], 
               pd.Series(['hello', 'hello world hello'])

array([[1, 0, 0],
   [1, 0, 1]])

I believe that the argument for this function should be a list with all the words and a series.

def words_in_texts(words, texts):
'''
Args:
    words (list-like): words to find
    texts (Series): strings to search in

Returns:
    NumPy array of 0s and 1s with shape (n, p) where n is the
    number of texts and p is the number of words.
'''
indicator_array = texts.str.contains(words)

return indicator_array

I'm confused on how to create the 2d array result, can anyone please help me with this? Thank you in advance!

MaxU - stand with Ukraine · Accepted Answer · 2018-04-21 18:27:51Z

2

Use sklearn.feature_extraction.text.CountVectorizer:

In [52]: from sklearn.feature_extraction.text import CountVectorizer

In [53]: vect = CountVectorizer(vocabulary=['hello', 'bye', 'world'], binary=True)

In [54]: X = vect.fit_transform(pd.Series(['hello', 'hello world hello']))

result as a sparse matrix:

In [55]: X
Out[55]:
<2x3 sparse matrix of type '<class 'numpy.int64'>'
        with 3 stored elements in Compressed Sparse Row format>

you can convert it to dense matrix:

In [56]: X.A
Out[56]:
array([[1, 0, 0],
       [1, 0, 1]], dtype=int64)

features (column names):

In [57]: vect.get_feature_names()
Out[57]: ['hello', 'bye', 'world']

edited Apr 21, 2018 at 18:27

answered Apr 21, 2018 at 18:25

MaxU - stand with Ukraine

212k37 gold badges402 silver badges437 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Ami Tavory Over a year ago

Ha ha, my link edit clashed with yours there. Good answer.

Collectives™ on Stack Overflow

How to create 2D array in python

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related