Python 之 Numpy (四)索引

代码：

import numpy as np

A = np.arange(3, 15).reshape((3, 4))
print(A)
# 索引到某一行
print(A[2])
# 索引某个元素
print(A[1][1])
print(A[1, 1])
print(A[1, :])  # 第一行所有元素
print(A[:, 1])  # 第一列的所有元素

运行结果：

[[ 3  4  5  6]
 [ 7  8  9 10]
 [11 12 13 14]]
[11 12 13 14]
8
8
[ 7  8  9 10]
[ 4  8 12]

代码：

# # 循环
# 输出每一行
for row in A:
    print("每一行")
    print(row)

运行结果：

每一行
[3 4 5 6]
每一行
[ 7  8  9 10]
每一行
[11 12 13 14]

代码：

# 输出每一列
for column in A.T:
    print(column)

运行结果：

[ 3  7 11]
[ 4  8 12]
[ 5  9 13]
[ 6 10 14]

代码：

# 输出每一个元素
print(A.flatten())
for item in A.flat:
    print(item)

运行结果：

[ 3  4  5  6  7  8  9 10 11 12 13 14]
3
4
5
6
7
8
9
10
11
12
13
14