我想為不同的城市和每個城市計算一些措施 5 次。我需要通過 bash 腳本提供數據。 #!/bin/bash cities=('paris''helsinki' 'rome') for city in "${cities[@]}" do for (( i = 1; i <= 5; i++ )) do srun python3 random_graph.py "$city" > "$city"/"city"_"$i".json done完畢我這樣寫我的python代碼： def get_network_measure(city): some code if __name__ == '__main__': city = input() result = get_network_measure(city)但我需要同時給出城市名稱，而不是在第二步通過 input() 函數給出。當我在 bash 中運行時，我需要這樣的東西python mycode.py paris
2 回答
繁星淼淼
TA貢獻1775條經驗 獲得超11個贊
您需要讀取傳遞給解釋器的參數。這些在sys.argv.
您還需要寫入標準輸出，sys.stdout.?您可以默認使用以下方式執行此操作print：
import sys
def main(argv):
? ? city = argv[1]
? ? print(get_network_measure(city))
def get_network_measure(city):
? ? return 110
if __name__ == '__main__':
? ? main(sys.argv)
您還可以通過管道輸入使用sys.stdin.
慕斯王
TA貢獻1864條經驗 獲得超2個贊
Python 模塊argparse為解析命令行參數提供了一個方便的功能：
import argparse
def main():
? ? parser = argparse.ArgumentParser()
? ? parser.add_argument('-c', '--city',?
? ? ? ? ? ? ? ? ? ? ? ? help = 'name of the city',?
? ? ? ? ? ? ? ? ? ? ? ? required = True
? ? ? ? ? ? ? ? ? ? ? ? )
? ? ? ? ? ? ? ? ? ? ? ?
? ? args = parser.parse_args()
? ??
? ? result = get_network_measure(args.city)
在 bash 中你必須這樣做：
$ python mycode.py -c paris
添加回答
舉報
0/150
提交
取消