Computer >> Máy Tính >  >> Lập trình >> Python

Trả về tổng tích lũy của các phần tử mảng coi NaN là 0 nhưng thay đổi kiểu kết quả trong Python

Để trả về tổng tích lũy của các phần tử mảng trên một trục nhất định coi các NaN là 0, hãy sử dụng phương thức thenancumprod (). Tổng tích lũy không thay đổi khi gặp các NaN và các NaN dồn lại được thay thế bằng các số không. Zeros được trả về cho các lát hoàn toàn là NaN hoặc trống.

Tham số đầu tiên là mảng đầu vào. Tham số thứ 2 là trục mà tổng tích lũy được tính theo đó. Giá trị mặc định (Không có) là tính tổng trên mảng được làm phẳng. Tham số thứ 3 là kiểu của mảng được trả về và của bộ tích lũy trong đó các phần tử được cộng dồn. Nếu dtype không được chỉ định, nó sẽ mặc định là dtype của a, trừ khi a có một dtype số nguyên có độ chính xác nhỏ hơn số nguyên nền tảng mặc định. Trong trường hợp đó, số nguyên nền tảng mặc định được sử dụng. Tham số thứ 4 là mảng đầu ra thay thế để đặt kết quả. Nó phải có hình dạng cùng tên và độ dài bộ đệm như đầu ra mong đợi nhưng kiểu sẽ được ép nếu cần.

Các bước

Đầu tiên, hãy nhập thư viện được yêu cầu -

import numpy as np

Tạo một mảng numpy bằng phương thức array (). Chúng tôi đã thêm các phần tử kiểu int với nan -

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

Hiển thị mảng -

print("Our Array...\n",arr)

Kiểm tra các thứ nguyên -

print("\nDimensions of our Array...\n",arr.ndim)

Lấy Datatype -

print("\nDatatype of our Array object...\n",arr.dtype)

Để trả về tổng tích lũy của các phần tử mảng trên một trục nhất định coi các NaN là 0, hãy sử dụng phương thức thenancumprod (). Tổng tích lũy không thay đổi khi gặp các NaN và các NaN tích lũy được thay thế bằng số không -

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

Ví dụ

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))

Đầu ra

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