def alternatingDictionary(self, unSortedDic):
'''
Input is dictionary or panda series. This is for plotting data on a pie Chart. We want the values of big and small items to alternate so there is no labeling overlap issues.
returns list of keys that represent alternating order of values.
'''
if type(unSortedDic) != dict:
if not isinstance(
unSortedDic,
pd.core.series.Series): #if its not a series throw error
raise cexcep.WrongFormat("Input must be a dictionary")
unSortedDic = unSortedDic.to_dict(
) # if its series can convert to dictionary
sortedDic = OrderedDict(sorted(unSortedDic.items(),
key=lambda t: t[1]))
keys = sortedDic.keys()
numKeys = len(keys)
firstHalf = keys[:int(.5 * numKeys)]
secondHalf = keys[int(.5 * numKeys):]
secondHalf = secondHalf[::-1] #reverse order of second list
target = []
for i in range(len(firstHalf)): #put back together
target = target + [firstHalf[i]] + [secondHalf[i]]
if numKeys % 2 == 1: #if there is an odd number of keys then must add last element
target += [secondHalf[-1]]
return target
def getTickFontSize(self, numTicks):
'''
input: int, output float
helper function for picking appropriate font size for graphs with ticks
'''
if not isinstance(numTicks, int): #if its not a series throw error
raise cexcep.WrongFormat("Input must be an int")
return None
if numTicks < 15:
return 15
maxFont = 15
minFont = 4
fontSizeFunction = interp1d(
[14, 75],
[maxFont, minFont]) #maps linearly range in [14,75] to [15,4]
fontsize = float(fontSizeFunction(numTicks))
return fontsize
def getColors(self, num_colors):
'''
input is an integer, output is a list of distinct colors in rgb format.
'''
if not isinstance(num_colors, int):
raise cexcep.WrongFormat("Input must be a integer")
colors = []
try:
for i in np.arange(
0., 360.,
360. / num_colors): #possible problem when num_colors =0
hue = i / 360.
lightness = (50 + np.random.rand() * 10) / 100.
saturation = (90 + np.random.rand() * 10) / 100.
colors.append(colorsys.hls_to_rgb(hue, lightness, saturation))
return colors
except ZeroDivisionError:
print "We don't have any crime record on this college on this campus."
return None