Merge pull request #180 from clouds56/patch-2

add rand and np.nan*

Author: tmke8

numpy-stubs/__init__.pyi

@@ -517,6 +517,9 @@ def arange(range_: int, dtype: Type[_DType]) -> ndarray[_DType]: ...
 def arange(range_: float) -> ndarray[float64]: ...
 def ascontiguousarray(a: Any, dtype: Optional[Type[_DType]] = ...) -> ndarray: ...
 def copy(a: Any, order: Optional[str] = ...) -> ndarray: ...
+def cumsum(
+    a: ndarray[_DType], axis: Optional[int] = ..., dtype: Optional[Type[_DType]] = ...
+) -> ndarray[_DType]: ...
 def delete(
     arr: ndarray[_DType], object: Union[int, List[int], slice], axis: Optional[int] = ...
 ) -> ndarray[_DType]: ...
@@ -770,11 +773,6 @@ def min(a: ndarray[_DType], axis: None = ...) -> _DType: ...
 @overload
 def min(a: ndarray[_DType], axis: _AxesType, keepdims: bool = ...) -> ndarray[_DType]: ...
 def minimum(a: ndarray[_DType], b: ndarray[_DType]) -> ndarray[_DType]: ...
-@overload
-def nanstd(a: ndarray[_Float]) -> _Float: ...
-@overload
-def nanstd(a: ndarray[_Float], axis: _AxesType, keepdims: bool = ...) -> ndarray[_Float]: ...
-def nansum(a: ndarray[_DType]) -> ndarray[_DType]: ...
 def nan_to_num(
     x: ndarray[_DType],
     copy: bool = ...,
@@ -887,6 +885,14 @@ def where(condition: _ConditionType, x: float, y: float) -> ndarray[float64]: ..
 @overload
 def where(condition: _ConditionType) -> Tuple[ndarray[int64], ...]: ...
 
+#
+# nan series methods
+#
+nancumsum = cumsum
+nanmean = mean
+nanstd = std
+nansum = sum
+
 #
 # Saving methods
 #

numpy-stubs/random.pyi

@@ -71,6 +71,7 @@ def normal(loc: float, scale: float, size: Union[int, Tuple[int, ...]]) -> ndarr
 def permutation(size: int) -> ndarray[int64]: ...
 @overload
 def permutation(size: Iterable[_DType]) -> ndarray[_DType]: ...
+def rand(*args: int) -> ndarray[_Float]: ...
 def randn(*args: int) -> ndarray[_Float]: ...
 @overload
 def randint(low: int, high: int = ...) -> int64: ...