Python的Numpy数组np.array()基本用法详解（二）

本节主要讲授array获取元素、转置、重塑等方法

"""
@Date: 2022-01-21
@Author: 苏雨流丰
@lang: Python
@summary: 访问、获取np.array的元素
"""

• 导入numpy包

import numpy as np

• 初始化工作

np_34_list = [[1, 3, 5, 7],
              [2, 4, 6, 8],
              [1, 2, 5, 6]]
np_44_list = [[1, 3, 5, 7],
              [2, 4, 6, 8],
              [1, 2, 5, 6],
              [3, 7, 4, 8]]
np_array_34 = np.array(np_34_list)
np_array_44 = np.array(np_44_list)

• 获取一个、一行、一列、一块的元素

print(np_array_44[2, 2])    # 第3行第3列 5
print(np_array_44[0])       # 第一行 [1 3 5 7]
print(np_array_44[1:3, 0:2])    # 切取2行2列（注意到隐含规则总是不含末index）
"""
[[2 4]
 [1 2]]
"""
print(np_array_44[:, 0])    # 第一列 [1 2 1 3]
print(np_array_44[0])       # 第一行 [1 3 5 7]

• 数组转置T, 重塑reshape, 倒转reverse(?), 降维flatten

print(np_array_34.T)
"""
[[1 2 1]
 [3 4 2]
 [5 6 5]
 [7 8 6]]
"""
print(np_array_34.reshape(4, 3))
"""
[[1 3 5]
 [7 2 4]
 [6 8 1]
 [2 5 6]]
"""
print(np_array_34[::-1, ::-1])
"""
[[6 5 2 1]
 [8 6 4 2]
 [7 5 3 1]]
"""
print(np_array_34.flatten())    # [1 3 5 7 2 4 6 8 1 2 5 6]