Convert single dictionary to list of dictionaries in vectorize function for consistent DataFrame creation
Browse files
controllers/vectorizer.py
CHANGED
@@ -40,6 +40,9 @@ def vectorize(article):
|
|
40 |
print(article)
|
41 |
print(type(article))
|
42 |
article['id'] = str(article['id'])
|
|
|
|
|
|
|
43 |
df = pd.DataFrame(article)
|
44 |
df = df[['id','site','title','titleCN','category','author','content',
|
45 |
'publishDate','link']]
|
|
|
40 |
print(article)
|
41 |
print(type(article))
|
42 |
article['id'] = str(article['id'])
|
43 |
+
# df = pd.DataFrame(article)
|
44 |
+
if isinstance(article, dict):
|
45 |
+
article = [article] # Convert single dictionary to list of dictionaries
|
46 |
df = pd.DataFrame(article)
|
47 |
df = df[['id','site','title','titleCN','category','author','content',
|
48 |
'publishDate','link']]
|