Вход командной строки сценария Python

Question

Для начала у меня мало знаний о python, так как я выполняю это задание как школьный проект. Мне нужно достичь следующего:

создать сценарий, который я могу начать с Windows CMD, и мне нужно добавить список параметров в первые 3 файла в каталоге.

я запустить скрипт следующим образом:

c:\Python34\python.exe c:\Python34\directory.py c:\temp 3 

содержание сценария:

##directory.py### 
import os 
import sys 

dirs = os.listdir(sys.argv[1]) 
print (dirs) 

Проблема в том, что я не думаю, что ключ командной строки «3» вводится в сценарии.

Любая идея, как я могу добиться этого?

Answer 1

Ваш код фактически не ссылается на вторую переменную командной строки. Вы пытаетесь перечислить первые n файлов в каталоге? Если да, попробуйте

import os 
import sys 

dir_arg=sys.argv[1] 
num_arg=int(sys.argv[2]) #need to convert second argument to a number 

dirs=os.listdir(dir_arg) 

print dirs[:num_arg] 

Answer 2

Как утверждают другие, sys.argv представляет собой просто набор аргументов командной строки, поэтому все, что вам нужно сделать, это изменить индекс, чтобы получить соответствующие значения.