Python NumPy 矩阵查询、定位与赋值操作详解

B = A[ A[:,6] == 5]

B = A[ A[:,6] > 5]

import numpy as np
#产生3行4列的矩阵
x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

y=x[1,1]
print(y)

import numpy as np
#产生3行4列的矩阵
x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

#[a:b,c:d] 表示取a-b行，c-d列。a、c可以省略，表示从0开始
#取x的前两行(所有列)
y=x[:2] # 等价于 y=x[:2,:]，等价于 y=x[0:2,...]
print(y)

#取x的前两列（的所有行）
z=x[:,:2] # 等价于 z=x[:,0:2]，等价于 z=x[...,0:2]
print(z)

import numpy as np
#产生3行4列的矩阵
x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

#[a:b,c:d] 表示取a-b行，c-d列。a、c可以省略，表示从0开始
#取x的第2行第2-3列
y=x[1,1:3]

import numpy as np
#产生3行4列的矩阵
x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

#  取出x中元素最大值所对应的索引，按照a[0][1]中的a[1]方向，即行方向搜索最大值。
y = np.argmax(x, 1)
print(y)

#  取出x中元素最大值所对应的行号
y = np.argmax(x, 0)
print(y)

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

# where返回一个长度为2的元组，第一个元素保存的是行号，第二个元素保存的是列号
y = np.where(x == np.max(x))
print(y)
print("最大值所在行：",y[0],"最大值所在列：",y[1])

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

y = np.sum(x, axis=1)
print(y)

xy = np.hstack((x, y.reshape(x.shape[0], -1)))
print(xy)

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

z = np.sum(x, axis=0)
print(z)

xz = np.vstack((x, z))
print(xz)

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

x[x<=5]=0 # 将小于5的数赋值为0
print(x)

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

# results = np.where(condition, x, y)
# 当条件为真时，对应位置返回x中的值，条件不成立则返回y中的值
y = np.where(x>5,x,0)  #满足大于5的值设置为x，不满足的设为0
print(y)

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

print(x>5) # x>5的为Ture，否则为False

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

x[1,1] = 999 # 指定行列号赋值
print(x)

x[1] = 123 # 指定行赋值
print(x)

x[:,1] = 321 # 指定列赋值
print(x)