numpy的索引

Posted jianglijian

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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)

  

以上是关于numpy的索引的主要内容,如果未能解决你的问题,请参考以下文章

快速 numpy 花式索引

在 numpy 数组中搜索元素的索引

Numpy 高级索引和基本索引

NumPy索引切片

python numpy片段

numpy:布尔索引和内存使用