def writeToFile(array,filename,mode="w"): #”w"代表着每次运行都覆盖内容 with open(filename,mode) as file: if len(array.shape)==1: # 一维矩阵 for i in range(array.shape[0]): file.write(str(array[i]) +"\n") else:# 二维矩阵 for i in range(array.shape[1]): for j in range(array.shape[0]): file.write(str(array[j,i])+" ") file.write("\n") file.close()

这段代码是用来将数组写入文件中的函数。它的参数包括一个数组、一个文件名和一个模式，其中模式默认为“w”，表示每次写入都会覆盖原有内容。该函数会根据数组的维度来判断是一维还是二维矩阵，并将其写入文件中。如果是一维矩阵，每个元素占一行；如果是二维矩阵，每行的元素用空格隔开，每行占一行。最后，函数会关闭文件。