I'm new to Python, and as I was coding for Project Euler to develop my skills in this language, I needed a function for converting between two bases. I know that int() can convert anything to base 10, but before I looked for a Python module concerning this math, I thought it would be a good idea to try and code one myself. Please, let me know if there is anything inefficient or in bad style so that I can improve.
It can handle A-Z notation (input both lower- and uppercase), any bases between 2 and 36, and catches problems related in invalid bases.
""" Matt Laporte <email@example.com> 2008 --- Base conversion tools. Allows for A-Z notation for digits greater than 9. """ from types import StringType def inBase(num, base, fromBase = 10, result = ''): """ Converts any number in fromBase to its representation in base. Defaults to converting to base_num from base_10. """ if result == '': #First call checks and corrections. if num == 0: return '0' base = __check(base) if fromBase != 10: fromBase = __check(fromBase) """ In order to maintain the simplicity of the actual conversion, this converts the number to base_10. """ num = toBase10(num, fromBase) fromBase = 10 if num < 0: #Further simplification for the conversion component. result = '-' + result num = -1 * num if num == 0: return result #Return the final result when num has been handled. else: #Conversion. This builds up result and eats away at num. thisDigit = num%base if thisDigit > 9: thisDigit = chr(thisDigit + 55) #Accounting for A-Z digits. else: thisDigit = str(thisDigit) return inBase(num/base, base, fromBase, thisDigit + result) def __check(base): """ Verifies that a given base is 2-9 or A-Z. Converts any base from A-Z to 10-35 for numerical calculations. Should be used in the form: base == __check(base). """ if type(base) == StringType and ((ord(base) > 96) and (ord(base) < 123)): #Correct a-z to A-Z base = chr(ord(base)-32) if type(base) == StringType and ((ord(base) > 64) and (ord(base) < 91)): #Correct A-Z to 10-35 return (ord(base) - 55) elif type(base) != type(1) or type(base) == StringType: #Not an integer, or any string other than A-Z. raise TypeError, 'invalid base type for inBase()' if base <= 1 or base > 36: raise ValueError, 'invalid base for inBase(): %s' % base return base def toBase10(num, base): """ Converts any number represented in base to its base_10 representation. """ sum = 0 parseNum = str(num) indices = range(len(parseNum)) #Exponents depend on digit place, indices necessary. for i in indices: #Add the decimal representations of the digits. sum += int(parseNum[i])*base**int(indices[::-1][i]) return sum
p.s. I haven't found the Python standard library function for any of this yet. As well, having a little experience in C, C++, Java, etc, I can say that Python is amazing. This -- and I know to be careful -- mutability and interchangeability (that is heresy in C) makes everything so easy in Python.
Edit: I used timeit. Although toBase10 was an attempt to imitate int() for the purpose of experiment, it is 10-15 times slower, so using int() in place of toBase10: 1 million executions of inBase with fromBase = 10 took around 10 seconds, with fromBase != 10 took around 20. I think this is pretty bad...