#!/opt/local/bin/python2
import numpy
import types
def foo(x):
""" foo """
if isinstance(x, float):
print 'float'
elif isinstance(x, complex):
print 'complex'
elif isinstance(x, int):
print 'int'
elif isinstance(x, list):
print 'list'
elif isinstance(x, str):
print 'string'
elif isinstance(x, numpy.ndarray):
if isinstance(x[0], types.FloatType):
print 'float array'
elif isinstance(x[0], types.IntType):
print 'int array'
elif isinstance(x[0], types.ComplexType):
print 'complex array'
else:
print('are')
else:
print 'error'
if __name__ == "__main__":
foo(1)
foo(1.0)
foo(1.0+2.0*1j)
foo([1,2,3])
foo('Hello')
foo(numpy.array([1,2,3]))
foo(numpy.array([1+1j,2+2j,3+3j]))
|