python数据处理

打开文件

f = open('C:\\Users\\Tesla\\Desktop\\pjxinjiang30')     #打开文件
reader = list(csv.reader(f))                            #读取文件、并将其转化为二维list
arr = np.array(reader)                                  #转换为矩阵
date = arr.T											#转置

定义空矩阵

mat = np.zeros(shape=(0,5)) 

范围内按条件查找

for i in range(20200716,20200804):                      #查找范围
    condition = (date == ' ' + str(i) + ' ')             #分类条件
    buffer = list(np.where(condition))
    if len(buffer[0]) != 0:                             
        buffer = arr[buffer[0][0] : buffer[0][len(buffer[0]) - 1] + 1]    #矩阵切片,筛选
        buffer = np.random.permutation(buffer)          #随机排序该条件下信息
        result_mat = np.r_[result_mat,buffer]           #信息追加到结果矩阵

新建index

result_mat = np.delete(result_mat,0,axis = 1)           #删除原有index
index = (np.arange(0,len(reader), 1)).reshape(len(reader),1)       #新建index
result_mat = np.c_[index,result_mat]

将矩阵写入文件

file = open('C:\\Users\\Tesla\\Desktop\\data','w')      #文件输出
file.write(str(result_mat)); 
file.close() 

解决numpy写入文件出现省略号

np.set_printoptions(threshold = np.inf) 

numpy写入文件宽度（自动换行）

np.set_printoptions(linewidth=1000)