返回数组元素的累积和,将NaN视为零,但更改Python中的结果类型
要返回在给定轴上对数组元素的累积和,并将NaN视为零,请使用nancumprod()方法。当遇到NaN时,累积和不会改变,前导NaN将被零替换。对于全是NaN或空的切片,将返回零。
第一个参数是输入数组。第二个参数是计算累积和的轴。默认值(None)是在展平的数组上计算累积和。第三个参数是返回数组和累加器的类型,在其中对元素求和。如果未指定dtype,则默认为a的dtype,除非a的整数dtype的精度小于默认平台整数的精度。在这种情况下,将使用默认平台整数。第四个参数是放置结果的备用输出数组。它必须与预期输出具有相同的形状和缓冲区长度,但如果需要,类型将被转换。
步骤
首先,导入所需的库:
import numpy as np
使用array()方法创建一个numpy数组。我们添加了包含nan的int类型元素:
arr = np.array([[10, 20, 30], [40, np.nan, 60]])
显示数组:
print("Our Array...\n",arr)
检查维度:
print("\nDimensions of our Array...\n",arr.ndim)
获取数据类型:
print("\nDatatype of our Array object...\n",arr.dtype)
要返回在给定轴上对数组元素的累积和,并将NaN视为零,请使用nancumprod()方法。当遇到NaN时,累积和不会改变,前导NaN将被零替换:
print("\nCumulative Sum of array elements...\n",np.nancumsum(arr, axis = 1, dtype = int))
示例
import numpy as np # Creating a numpy array using the array() method # We have added elements of int type with nan arr = np.array([[10, 20, 30], [40, np.nan, 60]]) # Display the array print("Our Array...\n",arr) # Check the Dimensions print("\nDimensions of our Array...\n",arr.ndim) # Get the Datatype print("\nDatatype of our Array object...\n",arr.dtype) # To return the cumulative sum of array elements over a given axis treating NaNs as zero, use the nancumprod() method print("\nCumulative Sum of array elements...\n",np.nancumsum(arr, axis = 1, dtype = int))
输出
Our Array... [[10. 20. 30.] [40. nan 60.]] Dimensions of our Array... 2 Datatype of our Array object... float64 Cumulative Sum of array elements... [[ 10 30 60] [ 40 40 100]]
广告