numpy之数据的切片

一维数组的索引

与列表，元组类似

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的列数据，即第一列和第三列
                                 ['刀妹' '艾克']]