Modified as requested to pass pep8's esthetics

pd3 · pd3 · commit 74e2ef5c6b44 · 2016-07-23T11:27:08.000+02:00
diff --git a/examples/statistics/customized_violin_demo.py b/examples/statistics/customized_violin_demo.py
@@ -5,48 +5,57 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
+
 # functions to calculate percentiles and adjacent values
-def percentile(vals,p):
+def percentile(vals, p):
     N = len(vals)
     n = p*(N+1)
     k = int(n)
     d = n-k
-    if k<=0: return vals[0]
-    if k>=N: return vals[N-1]
+    if k <= 0:
+        return vals[0]
+    if k >= N:
+        return vals[N-1]
     return vals[k-1] + d*(vals[k] - vals[k-1])
 
+
 def adjacent_values(vals):
-    q1 = percentile(vals,0.25)
-    q3 = percentile(vals,0.75)
+    q1 = percentile(vals, 0.25)
+    q3 = percentile(vals, 0.75)
     uav = q3 + (q3-q1)*1.5
-    if uav > vals[-1]: uav = vals[-1]
-    if uav < q3: uav = q3
+    if uav > vals[-1]:
+        uav = vals[-1]
+    if uav < q3:
+        uav = q3
     lav = q1 - (q3-q1)*1.5
-    if lav < vals[0]: lav = vals[0]
-    if lav > q1: lav = q1
-    return [lav,uav]
+    if lav < vals[0]:
+        lav = vals[0]
+    if lav > q1:
+        lav = q1
+    return [lav, uav]
 
 # create test data
 dat = [np.random.normal(0, std, 100) for std in range(6, 10)]
-lab = ['a','b','c','d']    # labels
+lab = ['a', 'b', 'c', 'd']    # labels
 med = []    # medians
 iqr = []    # inter-quantile ranges
 avs = []    # upper and lower adjacent values
 for arr in dat:
     sarr = sorted(arr)
-    med.append(percentile(sarr,0.5))
-    iqr.append([percentile(sarr,0.25),percentile(sarr,0.75)])
+    med.append(percentile(sarr, 0.5))
+    iqr.append([percentile(sarr, 0.25), percentile(sarr, 0.75)])
     avs.append(adjacent_values(sarr))
 
 # plot the violins
-fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(7,5))
-parts = ax.violinplot(dat,showmeans=False,showmedians=False,showextrema=False)
+fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(7, 5))
+parts = ax.violinplot(dat, showmeans=False, showmedians=False,
+                      showextrema=False)
 
 # plot medians and averages
 for i in range(len(med)):
-    ax.plot([i+1,i+1],avs[i],'-',c='black',lw=1)
-    ax.plot([i+1,i+1],iqr[i],'-',c='black',lw=5)
-    ax.plot(i+1,med[i],'o',mec='none',c='white',ms=6)
+    ax.plot([i+1, i+1], avs[i], '-', c='black', lw=1)
+    ax.plot([i+1, i+1], iqr[i], '-', c='black', lw=5)
+    ax.plot(i+1, med[i], 'o', mec='none', c='white', ms=6)
 
 # customize colors
 for pc in parts['bodies']:
@@ -58,13 +67,11 @@ def adjacent_values(vals):
 ax.xaxis.set_ticks_position('bottom')
 ax.set_xticks([x+1 for x in range(len(lab))])
 ax.set_xticklabels(lab)
-ax.set_xlim(0.25,len(lab)+0.75)
+ax.set_xlim(0.25, len(lab)+0.75)
 ax.set_ylabel('ylabel')
 ax.set_xlabel('xlabel')
 ax.set_title('customized violin plot')
 
 plt.subplots_adjust(bottom=0.15)
 
 plt.show()
-
-
