一维数组的索引
与列表,元组类似
import numpy as np
heros=np.array(["瑞文","寒冰","武器","老鼠","赏金"],dtype="U10")
print(heros[2])--------------->武器
一维数组的切片
获取后面两个
print(heros[3:])
print(heros[-2:])-------------->['老鼠' '赏金']
二维数组的索引
heros1=np.array([["风女","牛头","锤石"],
["刀妹","潘森","艾克"]],dtype="U5")
print(heros1.shape)------------->(2, 3)#查询数组类型,两行三列
获取牛头,即第一行第二列
print(heros1[0][1])
print(heros1[0,1])-------------->牛头
获取艾克,即第二行第三列
print(heros1[1][2])
print(heros1[1,2])-------------->艾克
二维数组的切片
获取某一行数据
print(heros1[0])--------------->['风女' '牛头' '锤石']
print(heros1[1])--------------->['刀妹' '潘森' '艾克']
获取第二列
print(heros1[:,1])------------->['牛头' '潘森']
获取第一列和第二列
print(heros1[:,:2])------------>[['风女' '牛头']# :2表示选取索引为0,1的列,即第一列和第二列
['刀妹' '潘森']]
获取第一列和第三列
print(heros1[:,::2])----------->[['风女' '锤石']# ::2表示步长为2,即选取步长为2的列数据,即第一列和第三列
['刀妹' '艾克']]