本关任务:编写计算城市间距离矩阵的函数。
时间: 2024-02-16 14:04:50 浏览: 117
好的,我可以帮您编写计算城市间距离矩阵的函数。请问您有城市坐标的数据吗?这里提供一个示例函数来计算城市间距离矩阵:
```python
import numpy as np
def distance_matrix(coords):
"""
计算城市间距离矩阵
参数:
coords: list, 包含各个城市的坐标信息,格式为[(x1, y1), (x2, y2), ..., (xn, yn)]
返回:
dist: numpy.array, 城市间距离矩阵,dist[i][j]表示第i个城市到第j个城市的距离
"""
n = len(coords)
dist = np.zeros((n, n))
for i in range(n):
for j in range(i+1, n):
dist[i][j] = dist[j][i] = np.sqrt((coords[i][0] - coords[j][0]) ** 2 + (coords[i][1] - coords[j][1]) ** 2)
return dist
```
这个函数接收一个包含城市坐标信息的列表`coords`,列表中每个元素是一个坐标元组`(x, y)`,函数返回一个numpy数组,表示城市间距离矩阵。
阅读全文