102 lines
2.5 KiB
Python
102 lines
2.5 KiB
Python
from collections import defaultdict
|
|
from itertools import product
|
|
|
|
|
|
def pipe(x):
|
|
return x
|
|
|
|
|
|
def print_pipe(x):
|
|
print(x)
|
|
return x
|
|
|
|
|
|
def flattenDict(l):
|
|
def tupalize(k, vs):
|
|
l = []
|
|
if isinstance(vs, list):
|
|
for v in vs:
|
|
l.append((k, v))
|
|
else:
|
|
l.append((k, vs))
|
|
return l
|
|
|
|
flat_list = [tupalize(k, vs) for k, vs in l.items()]
|
|
flat_dict = [dict(items) for items in product(*flat_list)]
|
|
return flat_dict
|
|
|
|
|
|
def flatten(l):
|
|
if isinstance(l, list):
|
|
return [item for sublist in l for item in sublist]
|
|
elif isinstance(l, dict):
|
|
return flattenDict(l)
|
|
|
|
|
|
def flatMap(f, collection):
|
|
return flatten(list(map(f, collection)))
|
|
|
|
|
|
def dict_filter(dictionary, condition):
|
|
return dict([(k, v) for k, v in dictionary.items() if condition(v)])
|
|
|
|
|
|
def contains_type(_collection, type):
|
|
return any(isinstance(x, type) for x in _collection)
|
|
|
|
|
|
def drop_right(l, n):
|
|
return l[:len(l) - n]
|
|
|
|
|
|
def key_filter(l, keyname):
|
|
return [v[keyname] for k, v in l.items()]
|
|
|
|
|
|
def groupByKey(l):
|
|
d = defaultdict(list)
|
|
for key, value in l:
|
|
d[key].append(value)
|
|
return list(dict(d).items()).pop()
|
|
|
|
|
|
# @curried
|
|
def rename(new_name, f):
|
|
f.__name__ = new_name
|
|
return f
|
|
|
|
|
|
def curry_pot(f, *argv):
|
|
sweep_ind = f.__name__[0:5] == 'sweep'
|
|
arg_len = len(argv)
|
|
if sweep_ind is True and arg_len == 4:
|
|
return f(argv[0])(argv[1])(argv[2])(argv[3])
|
|
elif sweep_ind is False and arg_len == 4:
|
|
return f(argv[0], argv[1], argv[2], argv[3])
|
|
elif sweep_ind is True and arg_len == 3:
|
|
return f(argv[0])(argv[1])(argv[2])
|
|
elif sweep_ind is False and arg_len == 3:
|
|
return f(argv[0], argv[1], argv[2])
|
|
else:
|
|
raise TypeError('curry_pot() needs 3 or 4 positional arguments')
|
|
|
|
# def curry_pot(f, *argv):
|
|
# sweep_ind = f.__name__[0:5] == 'sweep'
|
|
# arg_len = len(argv)
|
|
# if sweep_ind is True and arg_len == 4:
|
|
# return f(argv[0])(argv[1])(argv[2])(argv[3])
|
|
# elif sweep_ind is False and arg_len == 4:
|
|
# return f(argv[0])(argv[1])(argv[2])(argv[3])
|
|
# elif sweep_ind is True and arg_len == 3:
|
|
# return f(argv[0])(argv[1])(argv[2])
|
|
# elif sweep_ind is False and arg_len == 3:
|
|
# return f(argv[0])(argv[1])(argv[2])
|
|
# else:
|
|
# raise TypeError('curry_pot() needs 3 or 4 positional arguments')
|
|
|
|
# def rename(newname):
|
|
# def decorator(f):
|
|
# f.__name__ = newname
|
|
# return f
|
|
# return decorator
|