Python读入txt中的数组(下三角矩阵）
最近写遗传算法作业要用老师给的数据进行测试，于是学习了一些方法……
txt数据格式
提供的数据为城市间距离的下三角矩阵，直观显示大概为如下形式：
0
633 0
257 390 0
91 661 228 0
412 227 169 383 0
150 488 112 120 267 0
80 572 196 77 351 63 0
134 530 154 105 309 34 29 0
259 555 372 175 338 264 232 249 0
505 289 262 476 196 360 444 402 495 0
353 282 110 324 61 208 292 250 352 154 0
324 638 437 240 421 329 297 314 95 578 435 0
70 567 191 27 346 83 47 68 189 439 287 254 0
211 466 74 182 243 105 150 108 326 336 184 391 145 0
268 420 53 239 199 123 207 165 383 240 140 448 202 57 0
246 745 472 237 528 364 332 349 202 685 542 157 289 426 483 0
121 518 142 84 297 35 29 36 236 390 238 301 55 96 153 336 0
步骤
1. 使用np.loadtxt()读入txt文件
data = np.loadtxt('数据.txt', dtype=np.int, delimiter=' ')
这个方法就需要txt文件格式较为规范。
2. 自定义方法读入txt文件
def file2array(path, delimiter=' '):
fp = open(path, 'r', encoding='utf-8')
string = fp.read()
fp.close()
row_list = string.splitlines()
data_list = [[float(i) for i in row.strip().split(delimiter)] for row in row_list]
data = list(itertools.chain.from_iterable(data_list))
return data
3. 处理下三角矩阵
为了方便使用把空的部分填上。
distance = np.zeros([cityNum,cityNum])
for i in range(cityNum):
for j in range(cityNum):
distance[i][j] = data[p]
distance[j][i] = data[p]
p += 1
if data[p-1] == 0:
break
4. 将处理好的矩阵输出为txt文件
np.savetxt("test.txt", distance, fmt="%d", delimiter=" ")
效果如下：
总结
说实话也不知道为什么要折腾那么几下 = = 总之就是虽然都是基本操作但本人很怕麻烦……
参考资料
Original: https://blog.csdn.net/qq_43473795/article/details/125816296
Author: rthete
Title: Python读入txt中的数组
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/758030/
转载文章受原作者版权保护。转载请注明原作者出处！