Использование аргумента argv в конце совпадения строки - PullRequest
0 голосов
/ 15 октября 2018

Я делаю относительно простую задачу: взять строки внутри столбца данных и использовать эти строки для заполнения нового столбца.

Сначала я создаю новый столбец, а затем заполняю его обновленным именем Книги на основе строки, найденной в столбце «Имя книги» в кадре данных.Мне нужно соответствовать точную строку, включая год и квартал.

Мой старый код:

#Create empty column to map book names to

datagrouped["Book"] = ""

#Map book names by exact string matching the old values    

data['Book'][df['Book Name'] == America [Q3 2018]'] = North America 
data['Book'][df['Book Name'] == International [Q3 2018]'] = Intl 
data['Book'][df['Book Name'] == Asia [Q3 2018]'] = APAC

В идеале я хотел бы использовать argv, так как я запускаю это из командной строки.Я уже использую argv ранее в коде для чтения в файле CSV, и это работает.Мой вопрос: как бы я изменил это, чтобы принять другой argv (sys.argv [2]), чтобы я мог указать год и квартал в качестве аргумента в командной строке?До сих пор я пробовал

quarter = sys.argv[2] 

datagrouped["Book"] = ""

data['Book'][df['Book Name'] == America + str.quarter] = North America 
data['Book'][df['Book Name'] == International + str.quarter] = Intl
data['Book'][df['Book Name'] == Asia + str.quarter] = APAC 

Если есть что-то, что я могу добавить к этому вопросу, пожалуйста, дайте мне знать, и я обновлю его.

...