From b0a66da4deca32d0fb677d8ad38623b63902a319 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?D=C3=A9borah=20Servili?= Date: Thu, 21 Jul 2016 10:06:47 +0200 Subject: [PATCH] handling some NaN exceptions --- examples/situational-awareness/tools.py | 18 ++++-------------- 1 file changed, 4 insertions(+), 14 deletions(-) diff --git a/examples/situational-awareness/tools.py b/examples/situational-awareness/tools.py index f259f9c..57ee710 100644 --- a/examples/situational-awareness/tools.py +++ b/examples/situational-awareness/tools.py @@ -3,6 +3,7 @@ import json from json import JSONDecoder +import math import random import pygal from pygal.style import Style @@ -57,7 +58,7 @@ def toDatetime(date): ################ Formatting ################ def eventsListBuildFromList(filename): - with open('testt', 'r') as myfile: + with open(filename, 'r') as myfile: s=myfile.read().replace('\n', '') decoder = JSONDecoder() s_len = len(s) @@ -92,7 +93,7 @@ def eventsListBuildFromArray(filename): data.append(pd.DataFrame.from_dict(e, orient='index')) Events = pd.concat(data) for it in range(Events['attribute_count'].size): - if Events['attribute_count'][it] == None: + if Events['attribute_count'][it] == None or (isinstance(Events['attribute_count'][it], float) and math.isnan(Events['attribute_count'][it])): Events['attribute_count'][it]='0' else: Events['attribute_count'][it]=int(Events['attribute_count'][it]) @@ -127,20 +128,9 @@ def selectInRange(Events, begin=None, end=None): temp = Events.columns.tolist() inRange.columns = temp return inRange -''' -def isTagIn(dataframe, tag): - print 'tag =' + tag - result = [] - for tagname in dataframe['name']: - print tagname - if tag in tagname: - print 'True' - result.append(tagname) - return result -''' def isTagIn(dataframe, tag): - temp = Tags[Tags['name'].str.contains(test)].index.tolist() + temp = dataframe[dataframe['name'].str.contains(tag)].index.tolist() index = [] for i in range(len(temp)): if temp[i][0] not in index: