-
Notifications
You must be signed in to change notification settings - Fork 1
/
csv_to_df_pos.py
43 lines (30 loc) · 1.49 KB
/
csv_to_df_pos.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
import pandas as pd
import json
from io import StringIO
pd.set_option('display.max_rows',None)
pd.set_option('display.max_colwidth', 5000)
import stanza
#THE CHUNK BELOW IS ONLY IF YOU NEED TO MAKE A NEW CSV FILE IF YOU WANT TO CHANGE SIZE
#THIS WILL MAKE CSV FILE with 1million reviews
# with open('add directory of your json file') as f: #change to file directory for the json file
# chunks = pd.read_json(f, lines=True, chunksize=1000000)
# x=0
# # while x<10:
# for c in chunks:
# data = c.to_csv('Export1.csv',index = False)
# break
# #x+=1
col_list = ["reviewText"] #specify which columns you want to load into the dataframe from csv
df = pd.read_csv("Export1.csv", usecols=col_list)
# nlp = stanza.Pipeline(lang='en', processors='tokenize', 'mwt', 'pos') #USE WHEN STANZA FINALLY IMPORTS
#REMOVE NUMBERS
df['reviewText'] = df['reviewText'].str.replace(r'\d+', '')
print(df['reviewText'].head(48))
stanza.download('en')
stanza_nlp = stanza.Pipeline('en')
nlp = stanza.Pipeline(lang='en', processors='tokenize, mwt, pos')
#
def pos_tag(review):
doc = nlp(review) #function directly from stanza POS tag
print(*[f'word: {word.text}\tupos: {word.upos}\txpos: {word.xpos}\tfeats: {word.feats if word.feats else "_"}' for sent in doc.sentences for word in sent.words], sep='\n')
pos_tag(df.to_string(columns=col_list, index=False,header=False,max_rows=5)) #still need to figure out how to store the output from this function