如何在Python中使用线性索引访问2D数组

import numpy as np

# 创建一个2D数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 使用flatten()函数将2D数组展开成一维数组
flatten_arr = arr.flatten()

# 使用ravel()函数将2D数组展开成一维数组
ravel_arr = arr.ravel()

print(flatten_arr)  # 输出：[1 2 3 4 5 6 7 8 9]
print(ravel_arr)    # 输出：[1 2 3 4 5 6 7 8 9]

import numpy as np

# 创建一个2D数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 使用flatten()函数将2D数组展开成一维数组
flatten_arr = arr.flatten()

# 使用线性索引来访问展开后的一维数组
element = flatten_arr[4]  # 获取索引为4的元素，即第二行第二列的元素

print(element)  # 输出：5