数组切片
- 选取第二行第一列的数字
# 导入numpy包 以np的形式
import numpy as np
# 使用arange的方式创建一个二维数组
a = np.arange(1,13).reshape((3,4))
print(a)
# 选取第二行第一列的数字
print(a[2,1])
array([[ 1, 2, 3, 4],
[ 5, 6, 7, 8],
[ 9, 10, 11, 12]])
10
- 选取第二行的数据
print(a[2,:])
array([ 9, 10, 11, 12])
选取第二行与第二三列的数字
print(a[2,2:4])
array([11, 12])
-迭代遍历每一行的数
for row in a:
print(row)
[1 2 3 4]
[5 6 7 8]
[ 9 10 11 12]
- 迭代遍历每一列数据
for column in a.T:
print(column)
[1 5 9]
[ 2 6 10]
[ 3 7 11]
[ 4 8 12]
- 迭代遍历数组的每一项
for item in a.flat:
print(item)
1
2
3
4
5
6
7
8
9
10
11
12
- 将数组组合成一维数组
print(a.flatten())
array([ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12])