def update_readme(): # run benchmark suite while capturing output; indent with Capturing() as bench_list: bn.bench() bench_list = [' ' + b for b in bench_list] # read readme cwd = os.path.dirname(__file__) readme_path = os.path.join(cwd, '../README.rst') with open(readme_path) as f: readme_list = f.readlines() readme_list = [r.strip('\n') for r in readme_list] # remove old benchmark result from readme idx1 = readme_list.index(' Bottleneck performance benchmark') idx2 = [i for i, line in enumerate(readme_list) if line == ''] idx2 = [i for i in idx2 if i > idx1] idx2 = idx2[1] del readme_list[idx1:idx2] # insert new benchmark result into readme; remove trailing whitespace readme_list = readme_list[:idx1] + bench_list + readme_list[idx1:] readme_list = [r.rstrip() for r in readme_list] # replace readme file os.remove(readme_path) with open(readme_path, 'w') as f: f.write('\n'.join(readme_list))
def update_readme(): # run benchmark suite while capturing output; indent with Capturing() as bench_list: bn.bench() bench_list = [" " + b for b in bench_list] # read readme cwd = os.path.dirname(__file__) readme_path = os.path.join(cwd, "../README.rst") with open(readme_path) as f: readme_list = f.readlines() readme_list = [r.strip("\n") for r in readme_list] # remove old benchmark result from readme idx1 = readme_list.index(" Bottleneck performance benchmark") idx2 = [i for i, line in enumerate(readme_list) if line == ""] idx2 = [i for i in idx2 if i > idx1] idx2 = idx2[1] del readme_list[idx1:idx2] # insert new benchmark result into readme; remove trailing whitespace readme_list = readme_list[:idx1] + bench_list + readme_list[idx1:] readme_list = [r.rstrip() for r in readme_list] # replace readme file os.remove(readme_path) with open(readme_path, "w") as f: f.write("\n".join(readme_list))
#!/usr/bin/env python import bottleneck bottleneck.bench()
#!/usr/bin/env python3 import bottleneck bottleneck.bench()
import numpy as np import bottleneck as bn a = np.array([1, 2, np.nan, 4, 5]) print(bn.nansum(a)) print(bn.move_median(a, window=2, min_count=1)) bn.bench() bn.push()