python使用数组作为索引遍历数组

觉得有用的话,欢迎一起讨论相互学习~

python使用数组作为索引遍历数组

import numpy as np

a=np.arange(0,5)
print(a)
# [0 1 2 3 4]
b=np.arange(0,10).reshape(5,2)
print(b)
# [[0 1]
#  [2 3]
#  [4 5]
#  [6 7]
#  [8 9]]

# 将一维数组作为二维数组的索引
c0=b[a][:,0]
print(c0)
# [0 2 4 6 8]
c1=b[a][:,1]
print(c1)
# [1 3 5 7 9]
c2=b[a][0]
print(c2)
# [0 1]
c3=b[a][4]
print(c3)
# [8 9]
c4=b[a][:4]
print(c4)
# [[0 1]
#  [2 3]
#  [4 5]
#  [6 7]]