numpy 基础

# -*- coding: UTF-8 -*-
import numpy
vactor=numpy.array([5,10,15,20])
matris=numpy.array([[5,10,15,20],[25,30,35,40]])
print(vactor)
print(matris)
print(type(vactor))
print(type(matris))
print(vactor.shape)
print(matris.shape)

输出

[ 5 10 15 20]
[[ 5 10 15 20]
 [25 30 35 40]]
<type 'numpy.ndarray'>
<type 'numpy.ndarray'>
(4L,)
(2L, 4L)

通过BOOL矩阵去定位元素

# numpy数组
vactor1=numpy.array([5,10,15,20])
matris1=numpy.array([
    [5,10,15,20],
    [25,30,35,40],
    [55,60,65,70]
])
print(vactor1[0:2])
print(matris1[0,2])
print(matris1[:,2])
print(matris1[:,0:2])#  :代表所有
print(matris1[1:3,0:2])# 2-3行，1-2列
#通过转化为bool去定位数据
vactor1_bool = (vactor1==10)
print(vactor1_bool)
print(vactor1[vactor1_bool])

matris1_bool=(matris1[:,1]==30)
print(matris1_bool)
print(matris1[matris1_bool,:])

输出：

[ 5 10]
15
[15 35 65]
[[ 5 10]
 [25 30]
 [55 60]]
[[25 30]
 [55 60]]
[False  True False False]
[10]
[False  True False]
[[25 30 35 40]]