NumPy ndarray.flat 函数

 NumPy 函数

NumPy ndarray.flat() 是一个数组属性，它是数组上的一维迭代器。

语法

numpy.ndarray.flat 

参数

不需要参数。

返回值

返回数组上的一维迭代器。

示例:

在下面的示例中，ndarray.flat 用于访问数组的元素。

import numpy as np

x = np.array([[10,20,30],
              [40,50,60]])
n = x.size

#使用 flat 属性显示所有元素
y1 = x.flat[0:n]
print("x.flat[0:n] =", y1)

#使用平面属性显示第三个元素
y2 = x.flat[3]
print("x.flat[3] =", y2)

print()
#使用平面属性显示所有元素
#on 数组转置
y3 = x.T.flat[0:n]
print("x.T.flat[0:n] =", y3)

#使用平面属性显示第三个元素
#on 数组转置
y4 = x.T.flat[3]
print("x.flat[3] =", y4) 

输出上述代码的内容为:

x.flat[0:n] = [10 20 30 40 50 60]
x.flat[3] = 40

x.T.flat[0:n] = [10 40 20 50 30 60]
x.flat[3] = 50 

示例:

ndarray.flat属性可用于为数组赋值。

import numpy as np

x = np.array([[10,20,30],
              [40,50,60]])
y = np.array([[10,20,30],
              [40,50,60]])

#给所有元素赋值
#使用平面属性的数组
x.flat = 10
print("x =")
print(x)

print()
#给给定元素赋值
#使用平面属性的数组
y.flat[[1, 2, 3]] = 5
print("y =")
print(y) 

上述代码的输出将是:

x =
[[10 10 10]
 [10 10 10]]

y =
[[10  5  5]
 [ 5 50 60]]