返回将 NaN 视为零的数组元素的累积和,但在 Python 中更改结果类型

pythonnumpyserver side programmingprogramming

要返回将 NaN 视为零的给定轴上的数组元素的累积和,请使用 nancumprod() 方法。当遇到 NaN 并将前导 NaN 替换为零时,累积和不会改变。对于全为 NaN 或为空的切片,将返回零。

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

步骤

首先,导入所需的库 −

import numpy as np

使用 array() 方法创建 numpy 数组。我们用 nan 添加了 int 类型的元素 −

arr = np.array([[10, 20, 30], [40, np.nan, 60]])

显示数组 −

print("我们的数组...\n",arr)

检查维度 −

print("\n数组的维度...\n",arr.ndim)

获取数据类型 −

print("\n我们的数组对象的数据类型...\n",arr.dtype)

要返回将 NaN 视为零的给定轴上的数组元素的累积和,请使用 nancumprod() 方法。遇到 NaN 时累积和不会改变,并且前导 NaN 会被零替换 −

print("\n数组元素的累计和...\n",np.nancumsum(arr, axis = 1, dtype = int))

示例

import numpy as np

# 使用 array() 方法创建 numpy 数组
# 我们添加了 int 类型的元素 with nan
arr = np.array([[10, 20, 30], [40, np.nan, 60]])

# 显示数组
print("我们的数组...\n",arr)

# 检查维度
print("\n数组的维度...\n",arr.ndim)

# 获取数据类型
print("\n我们的数组对象的数据类型...\n",arr.dtype)

# 要返回将 NaN 视为零的给定轴上的数组元素的累积和,请使用 nancumprod() 方法

print("\n数组元素的累计和...\n",np.nancumsum(arr, axis = 1, dtype = int))

输出

我们的数组...
[[10. 20. 30.]
[40. nan 60.]]

我们的数组的维度...
2

我们的数组对象的数据类型...
float64

数组元素的累计和...
[[ 10 30 60]
[ 40 40 100]]

相关文章