numpy的索引

import numpy as np

A =np.arange(3,15).reshape(3,4)
print(A)
#第一行
print(A[2])
#返回元素
print(A[1][2])
print(A[1,2])
#返回列
print(A[:,1])
#返回行
print(A[2,:])
#返回一段元素
print(A[1,1:2])

#迭代每一行
for row in A:
    print(row)
#迭代每一列
for column in A.T:
    print(column)
#返回每一个元素到同一行
print(A.flatten())
#迭代每一个元素
for item in A.flat:
    print(item)

  

原文地址:https://www.cnblogs.com/jianglijian/p/10087495.html