在 Python 中返回数组元素在给定轴 1 上的累积和，并将 NaN 视为零

要返回数组元素在给定轴上累积和，并将 NaN 视为零，请使用 nancumprod() 方法。当遇到 NaN 时，累积和不会发生变化，并且前导 NaN 将被零替换。对于全是 NaN 或为空的切片，将返回零。该方法返回一个包含结果的新数组，除非指定了 out，在这种情况下，它将返回 out。结果的大小与 a 相同，如果 axis 不为 None 或 a 为一维数组，则形状与 a 相同。

第一个参数是输入数组。第二个参数是计算累积和的轴。默认值（None）是在展平的数组上计算累积和。第三个参数是返回数组和累加器中元素求和的类型的类型。如果未指定 dtype，则默认为 a 的 dtype，除非 a 的整数 dtype 的精度小于默认平台整数的精度。在这种情况下，将使用默认平台整数。第四个参数是放置结果的备用输出数组。它必须与预期输出具有相同的形状和缓冲区长度，但类型将在必要时进行转换。

步骤

首先，导入所需的库 -

import numpy as np

使用 array() 方法创建一个 NumPy 数组。我们添加了 int 类型和 nan 的元素 -

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() 方法 -

print("\nCumulative Sum of array elements...\n",np.nancumsum(arr, axis = 1))

示例

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
# The cumulative sum does not change when NaNs are encountered and leading NaNs are replaced by zeros.
print("\nCumulative Sum of array elements...\n",np.nancumsum(arr, axis = 1))

输出

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.]]

AmitDiwan

更新于： 2022年2月24日

99 次浏览

启动您的 职业生涯

通过完成课程获得认证

开始