numpy数组 传入 int 和 传入 list[] 的区别.pdfVIP

numpy数数组组传传入入int和和传传入入list[]的的区区别别

pipinstallbpython

$bpython

bash命令行运行，直接上代码

importnumpyasnp

a=np.array([[1,2,3],[3,4,5],[111,222,333],[333,0,0]])

print(a)

[[123]

[345]

[111222333]

[33300]]

#传入int

print(a[1][0])

3

#传入list

print(a[[1]][0])

[345]

#可以看到，numpy允许传入[]来取对应行

#取第[1,2,3]行得到一新的numpy数组

print(a[[1,2,3]])

[[345]

[111222333]

[33300]]

print(a[[1,2,3]][0])

[345]

print(a[[1,2,3]][1])

[111222333]

print(a[[1,2,3]][2])

[33300]