A python function to memoize DP problems with multiple-index keys

memoize_many.py

def memoize(*indices):
	def decorator(function):
		cache = {}
		def proxy(*args, **kwargs):
			key_args = [args[i] for i in indices] if indices else args
			key = tuple(key_args)
			if key not in cache:
				cache[key] = function(*args, **kwargs)
			return cache[key]
		return proxy
	return decorator