Как в консоли передать скрипту 2 файла в качестве аргумента?

Question

[rmfalx]

Задача такая - есть 2 файла cvs. Нужно написать скрипт на питоне, который при помощи pandas объединят эти таблицы по совпадающим id в обеих таблицах и фильтровать по определенным полям.

файлы cvs каждый раз будут разные.

Интерфейс такой. В консоли запускаем скрипт

$python script.py имя.файла1 имя.файла2

После этого скрипт выполняем объединение и фильтрацию.

Подскажите, как передать в скрипт файлы как аргументы, а так же как объединить таблицы. Я новичок в питоне.

Спасибо.

Answer 1

[Anton Kuzmichev]

Ну, вы уже передали имена файлов в скрипт как аргументы. Осталось научиться принимать их.

import sys
print(sys.argv)

sys.argv - список переданных аргументов, причём первым идёт имя самого исполняемого скрипта.

Comments

[rmfalx]

Ага, он принял их

Args: ['script_csv.py', 'report (27).csv', 'seotime_mails (1).csv']

А не подскажите, как теперь с ними работать?

import sys
from pandas import read_csv

df1 = read_csv(sys.argv[1])
df2 = read_csv(sys.argv[2])

$python script.py имя.файла1 имя.файла2

Так это будет работать?

[Anton Kuzmichev]

rmfalx: ну а как же. Главное правильно передать. Судя по вашему коду, эти файлы расположены в той же папке, где и скрипт, так что да.

Я новичок в питоне.

Не бойтесь сами запускать и пробовать.

[rmfalx]

Anton Kuzmichev: cтранно, я установил pandas через pip install, но он не работает

import pandas as pd
ImportError: No module named pandas

[Антон Б]

rmfalx: Вы используете Windows или Linux? В Linux по умолчанию может быть одновременно установлено две версии питона. Соответственно pip мог поставить модуль не для той версии через который вы свой скрипт запускаете.

[rmfalx]

Антон Б: я ставил для 3 питона через pip3 Install

[Антон Б]

rmfalx, А что собственно выдает pip3 install pandas?
Если уже установлена, то должно быть что то вроде
Requirement already satisfied: pandas in /usr/local/lib/python3.5/dist-packages
И убедитесь что запускаете именно интерпретатор именно третьего питона.

[rmfalx]

Requirement already satisfied: pandas in ./.local/lib/python3.5/site-packages
Requirement already satisfied: pytz>=2011k in ./.local/lib/python3.5/site-packages (from pandas)
Requirement already satisfied: numpy>=1.7.0 in ./.local/lib/python3.5/site-packages (from pandas)
Requirement already satisfied: python-dateutil>=2 in ./.local/lib/python3.5/site-packages (from pandas)
Requirement already satisfied: six>=1.5 in ./.local/lib/python3.5/site-packages (from python-dateutil>=2->pandas)

[rmfalx]

Антон Б,

В общем вот такая вот петрушка

x@x-desktop:~$ pip3 install pandas
Requirement already satisfied: pandas in ./.local/lib/python3.5/site-packages
Requirement already satisfied: pytz>=2011k in ./.local/lib/python3.5/site-packages (from pandas)
Requirement already satisfied: numpy>=1.7.0 in ./.local/lib/python3.5/site-packages (from pandas)
Requirement already satisfied: python-dateutil>=2 in ./.local/lib/python3.5/site-packages (from pandas)
Requirement already satisfied: six>=1.5 in ./.local/lib/python3.5/site-packages (from python-dateutil>=2->pandas)
x@x-desktop:~$ cd Загрузки/
x@x-desktop:~/Загрузки$ python script_csv.py report\ \(27\).csv seotime_mails\ \(1\).csv 
Traceback (most recent call last):
  File "script_csv.py", line 2, in <module>
    import pandas as pd
ImportError: No module named pandas

Сам код скрипта

import sys
import pandas as pd
from pandas import read_csv

df1 = read_csv(sys.argv[1])
df2 = read_csv(sys.argv[2])

print(df1)
print(df2)


print sys.argv

[Anton Kuzmichev]

rmfalx, что выдаёт команда python --version?

[Антон Б]

rmfalx, Я подозреваю, что вы запускаете второй питон. Попробуйте запустить скрипт командой python3.

[rmfalx]

Антон Б, Да блин! Спасибо огромное)

[rmfalx]

Вы крутые, я не внимательный! Всем спасибо!