村庄存入村庄数个,道路存入时存为列表集合,列表的元素为元组第一个位置和第二个位置是村庄名称第三个位置是道路长度,上述问题给出python代码
时间: 2024-02-22 13:57:15 浏览: 15
以下是一个示例代码,用于将村庄和道路存储为列表集合:
```
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--num', type=int, help='村庄数目')
parser.add_argument('--edge', type=int, help='可建公路数目')
parser.add_argument('--village', nargs='*', help='村庄名称')
parser.add_argument('--road', nargs='+', help='村村连同道路信息')
args = parser.parse_args()
villages = args.village
roads = []
for road_info in args.road:
road_info = road_info.split(',')
source = road_info[0]
destination = road_info[1]
length = int(road_info[2])
roads.append((source, destination, length))
print('村庄列表:', villages)
print('道路列表:', roads)
```
这段代码将--village和--road参数解析为villages和roads两个列表。其中,--village参数直接解析为列表;--road参数需要逐个解析为元组,并加入到列表中。在解析道路信息时,我们使用了split()方法将字符串按逗号切分为三个部分,然后将其转换为元组并加入到列表中。最后,我们将村庄列表和道路列表打印出来,以供参考。