本节主要讲授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]
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]