summaryrefslogtreecommitdiff
path: root/numpy/lib/user_array.py
diff options
context:
space:
mode:
authorCharles Harris <charlesr.harris@gmail.com>2014-07-30 18:06:28 -0600
committerJulian Taylor <jtaylor.debian@googlemail.com>2014-07-31 21:21:17 +0200
commit01b0d7e82211b581aaff925e3ccc36cff9ac1895 (patch)
tree8ec68353d5f09b9f0411948f1345ec79f5443b4c /numpy/lib/user_array.py
parentdec6658cdc10a23ad0e733fb52a814306033d88c (diff)
downloadnumpy-01b0d7e82211b581aaff925e3ccc36cff9ac1895.tar.gz
STY: Make files in numpy/lib PEP8 compliant.
The rules enforced are the same as those used for scipy.
Diffstat (limited to 'numpy/lib/user_array.py')
-rw-r--r--numpy/lib/user_array.py139
1 files changed, 99 insertions, 40 deletions
diff --git a/numpy/lib/user_array.py b/numpy/lib/user_array.py
index 3dde538d8..bb5bec628 100644
--- a/numpy/lib/user_array.py
+++ b/numpy/lib/user_array.py
@@ -7,29 +7,33 @@ complete.
from __future__ import division, absolute_import, print_function
from numpy.core import (
- array, asarray, absolute, add, subtract, multiply, divide,
- remainder, power, left_shift, right_shift, bitwise_and, bitwise_or,
- bitwise_xor, invert, less, less_equal, not_equal, equal, greater,
- greater_equal, shape, reshape, arange, sin, sqrt, transpose
- )
+ array, asarray, absolute, add, subtract, multiply, divide,
+ remainder, power, left_shift, right_shift, bitwise_and, bitwise_or,
+ bitwise_xor, invert, less, less_equal, not_equal, equal, greater,
+ greater_equal, shape, reshape, arange, sin, sqrt, transpose
+)
from numpy.compat import long
+
class container(object):
+
def __init__(self, data, dtype=None, copy=True):
self.array = array(data, dtype, copy=copy)
def __repr__(self):
if len(self.shape) > 0:
- return self.__class__.__name__+repr(self.array)[len("array"):]
+ return self.__class__.__name__ + repr(self.array)[len("array"):]
else:
- return self.__class__.__name__+"("+repr(self.array)+")"
+ return self.__class__.__name__ + "(" + repr(self.array) + ")"
- def __array__(self,t=None):
- if t: return self.array.astype(t)
+ def __array__(self, t=None):
+ if t:
+ return self.array.astype(t)
return self.array
# Array as sequence
- def __len__(self): return len(self.array)
+ def __len__(self):
+ return len(self.array)
def __getitem__(self, index):
return self._rc(self.array[index])
@@ -37,19 +41,21 @@ class container(object):
def __getslice__(self, i, j):
return self._rc(self.array[i:j])
-
def __setitem__(self, index, value):
self.array[index] = asarray(value, self.dtype)
+
def __setslice__(self, i, j, value):
self.array[i:j] = asarray(value, self.dtype)
def __abs__(self):
return self._rc(absolute(self.array))
+
def __neg__(self):
return self._rc(-self.array)
def __add__(self, other):
- return self._rc(self.array+asarray(other))
+ return self._rc(self.array + asarray(other))
+
__radd__ = __add__
def __iadd__(self, other):
@@ -57,32 +63,40 @@ class container(object):
return self
def __sub__(self, other):
- return self._rc(self.array-asarray(other))
+ return self._rc(self.array - asarray(other))
+
def __rsub__(self, other):
- return self._rc(asarray(other)-self.array)
+ return self._rc(asarray(other) - self.array)
+
def __isub__(self, other):
subtract(self.array, other, self.array)
return self
def __mul__(self, other):
return self._rc(multiply(self.array, asarray(other)))
+
__rmul__ = __mul__
+
def __imul__(self, other):
multiply(self.array, other, self.array)
return self
def __div__(self, other):
return self._rc(divide(self.array, asarray(other)))
+
def __rdiv__(self, other):
return self._rc(divide(asarray(other), self.array))
+
def __idiv__(self, other):
divide(self.array, other, self.array)
return self
def __mod__(self, other):
return self._rc(remainder(self.array, other))
+
def __rmod__(self, other):
return self._rc(remainder(other, self.array))
+
def __imod__(self, other):
remainder(self.array, other, self.array)
return self
@@ -90,59 +104,74 @@ class container(object):
def __divmod__(self, other):
return (self._rc(divide(self.array, other)),
self._rc(remainder(self.array, other)))
+
def __rdivmod__(self, other):
return (self._rc(divide(other, self.array)),
self._rc(remainder(other, self.array)))
def __pow__(self, other):
return self._rc(power(self.array, asarray(other)))
+
def __rpow__(self, other):
return self._rc(power(asarray(other), self.array))
+
def __ipow__(self, other):
power(self.array, other, self.array)
return self
def __lshift__(self, other):
return self._rc(left_shift(self.array, other))
+
def __rshift__(self, other):
return self._rc(right_shift(self.array, other))
+
def __rlshift__(self, other):
return self._rc(left_shift(other, self.array))
+
def __rrshift__(self, other):
return self._rc(right_shift(other, self.array))
+
def __ilshift__(self, other):
left_shift(self.array, other, self.array)
return self
+
def __irshift__(self, other):
right_shift(self.array, other, self.array)
return self
def __and__(self, other):
return self._rc(bitwise_and(self.array, other))
+
def __rand__(self, other):
return self._rc(bitwise_and(other, self.array))
+
def __iand__(self, other):
bitwise_and(self.array, other, self.array)
return self
def __xor__(self, other):
return self._rc(bitwise_xor(self.array, other))
+
def __rxor__(self, other):
return self._rc(bitwise_xor(other, self.array))
+
def __ixor__(self, other):
bitwise_xor(self.array, other, self.array)
return self
def __or__(self, other):
return self._rc(bitwise_or(self.array, other))
+
def __ror__(self, other):
return self._rc(bitwise_or(other, self.array))
+
def __ior__(self, other):
bitwise_or(self.array, other, self.array)
return self
def __pos__(self):
return self._rc(self.array)
+
def __invert__(self):
return self._rc(invert(self.array))
@@ -150,33 +179,62 @@ class container(object):
if len(self.shape) == 0:
return func(self[0])
else:
- raise TypeError("only rank-0 arrays can be converted to Python scalars.")
+ raise TypeError(
+ "only rank-0 arrays can be converted to Python scalars.")
+
+ def __complex__(self):
+ return self._scalarfunc(complex)
- def __complex__(self): return self._scalarfunc(complex)
- def __float__(self): return self._scalarfunc(float)
- def __int__(self): return self._scalarfunc(int)
- def __long__(self): return self._scalarfunc(long)
- def __hex__(self): return self._scalarfunc(hex)
- def __oct__(self): return self._scalarfunc(oct)
+ def __float__(self):
+ return self._scalarfunc(float)
- def __lt__(self, other): return self._rc(less(self.array, other))
- def __le__(self, other): return self._rc(less_equal(self.array, other))
- def __eq__(self, other): return self._rc(equal(self.array, other))
- def __ne__(self, other): return self._rc(not_equal(self.array, other))
- def __gt__(self, other): return self._rc(greater(self.array, other))
- def __ge__(self, other): return self._rc(greater_equal(self.array, other))
+ def __int__(self):
+ return self._scalarfunc(int)
- def copy(self): return self._rc(self.array.copy())
+ def __long__(self):
+ return self._scalarfunc(long)
- def tostring(self): return self.array.tostring()
+ def __hex__(self):
+ return self._scalarfunc(hex)
- def byteswap(self): return self._rc(self.array.byteswap())
+ def __oct__(self):
+ return self._scalarfunc(oct)
- def astype(self, typecode): return self._rc(self.array.astype(typecode))
+ def __lt__(self, other):
+ return self._rc(less(self.array, other))
+
+ def __le__(self, other):
+ return self._rc(less_equal(self.array, other))
+
+ def __eq__(self, other):
+ return self._rc(equal(self.array, other))
+
+ def __ne__(self, other):
+ return self._rc(not_equal(self.array, other))
+
+ def __gt__(self, other):
+ return self._rc(greater(self.array, other))
+
+ def __ge__(self, other):
+ return self._rc(greater_equal(self.array, other))
+
+ def copy(self):
+ return self._rc(self.array.copy())
+
+ def tostring(self):
+ return self.array.tostring()
+
+ def byteswap(self):
+ return self._rc(self.array.byteswap())
+
+ def astype(self, typecode):
+ return self._rc(self.array.astype(typecode))
def _rc(self, a):
- if len(shape(a)) == 0: return a
- else: return self.__class__(a)
+ if len(shape(a)) == 0:
+ return a
+ else:
+ return self.__class__(a)
def __array_wrap__(self, *args):
return self.__class__(args[0])
@@ -200,19 +258,20 @@ class container(object):
# Test of class container
#############################################################
if __name__ == '__main__':
- temp=reshape(arange(10000), (100, 100))
+ temp = reshape(arange(10000), (100, 100))
- ua=container(temp)
+ ua = container(temp)
# new object created begin test
print(dir(ua))
- print(shape(ua), ua.shape) # I have changed Numeric.py
+ print(shape(ua), ua.shape) # I have changed Numeric.py
- ua_small=ua[:3, :5]
+ ua_small = ua[:3, :5]
print(ua_small)
- ua_small[0, 0]=10 # this did not change ua[0,0], which is not normal behavior
+ # this did not change ua[0,0], which is not normal behavior
+ ua_small[0, 0] = 10
print(ua_small[0, 0], ua[0, 0])
- print(sin(ua_small)/3.*6.+sqrt(ua_small**2))
+ print(sin(ua_small) / 3. * 6. + sqrt(ua_small ** 2))
print(less(ua_small, 103), type(less(ua_small, 103)))
- print(type(ua_small*reshape(arange(15), shape(ua_small))))
+ print(type(ua_small * reshape(arange(15), shape(ua_small))))
print(reshape(ua_small, (5, 3)))
print(transpose(ua_small))