python numpy 矩阵操作

import numpy np

1. 读写数组,这里可以看成矩阵

#返回值格式(评分,信任表,用户个数,项目个数)
 a = np.arange(0,12,0.5).reshape(4,-1)
 np.savetxt("a.txt", a) # 缺省按照'%.18e'格式保存数据,以空格分隔
 np.loadtxt("a.txt")
 np.loadtxt('a.txt',dtype='int')#设置读出的数据类型

2. 使用数组的reshape方法,可以创建一个改变了尺寸的新数组,原数组的shape保持不变:

a = np.arange(4).reshape(2,2)
b = np.arange(6).reshape(2,3)
print('the result is\n ',a)
print('the result is\n ',b)

3.transpose()对数组进行转置

print('the transpose is\n ',b.transpose())#转置

4. 矩阵运算

np.dot() 矩阵的乘法
print('the result is\n ',np.dot(a,b))#矩阵乘
np.linalg.inv() 求矩阵的逆
print('the inverse is\n ',np.linalg.inv(a))#逆矩阵

5. 求行列大小

(m,n) = a.shape#求行列个数

6. 求最值

temp1 = np.max(a[:,0])
temp2 = np.min(a[:,0])

7. 求第三列等于1的个数

np.sum(a[:,2]==1)

8. 
求一组随机数组

randIndices = np.random.permutation(4)

ans=array[[3,0,2,1]]

9. 组合两个数组

np.vstack((a,b))#纵向结合,保证列数相同 注意双括号
np.hstack((a,b))#横向结合,保证行数相同

10. 求和 和 计算平均数

np.sum(a,0)#0代表求得是各列的和

np.mean(a,1)#1代表求得各行的平均数

11.求交集

np.intersectld(a[0,:],b[0,:])#求两个一维数组的交集

12.条件查询

np.where(a>5)#找到所有满足条件的位置

np.where(a>5,0,1)#找到这些值满足赋值为0,不满足赋值为1

13.矩阵写文件 追加 方式

fileObject = open('result.txt','a')#追加的方式打开文件
a = [[1,2,3] ,[3,4,5] , [4,5,6]]#存取list
for i in a:
  tmp = str(i)
  tmp = tmp.replace('[','')
  tmp = tmp.replace('[','')+'\n'
  fileObject.weite(tmp)
fileObject.close()
b = np.loadtxt('result.txt',delimiter=',')#同样可以读出矩阵
print(b[:,:])

你可能感兴趣的:(python)