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

演化计算与人工智能

发布于 2020-08-14 00:15:24

2.6K00

代码可运行

文章被收录于专栏：人工智能与演化计算成长与进阶人工智能与演化计算成长与进阶

运行总次数：0

代码可运行

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]]

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2020-06-14，如有侵权请联系 cloudcommunity@tencent.com 删除

python