Numpy MaskedArray.flatten()函数 | Python

在 NumPy 中，生成随机数组时常常会有一些缺失值。这些缺失值可以用 NumPy 的 MaskedArray 类型处理。MaskedArray.flatten() 函数将一个带缺失值的 n 维数组展平为一个一维数组。

语法

numpy.ma.flatten(arr, order='C')

参数

• arr: 待展平的带缺失值的 NumPy 数组。
• order: （可选参数）字符串或用于排序的元组。可选值为 'C'（默认）和 'F'。

返回值

返回一个一维数组，将带缺失值的 n 维数组展平为一个一维数组。

示例

import numpy as np

a = np.ma.array([[1, 2], [3, 4]], mask=[[False, True], [False, False]])
print("展平前：")
print(a)
print("展平后：", a.flatten())

输出：

展平前：
[[1 --]
 [3 4]]
展平后： [1. 3. 4.]

解释

在上面的示例中，我们创建了一个带有缺失值的二维数组。在这个数组中，第一个元素没有缺失值，但是第二个元素缺失了。

使用 MaskedArray.flatten() 函数将带缺失值的二维数组展平为一维数组。展平后，缺失值将被自动去掉，原先的二维数组将被展平为一个只有一维的数组。