Задача курса
pandas .dt: год, месяц, день недели и фильтрация
Курс «NumPy и pandas: практический тренажёр» · урок «Временные столбцы: парсинг, .dt-компоненты, длительности и фильтрация»
Закрепите работу с datetime-столбцом и фильтрацию таблицы по вычисленным компонентам даты.
Условие
Архив заказов: год, месяц, день недели и фильтрация по месяцу
Что уже дано
Проверяющая система запускает вашу программу и подаёт данные через stdin. Никаких заранее созданных переменных нет — вы читаете всё сами.
Что нужно сделать
Напишите программу, которая обрабатывает архив заказов:
- Читает из
stdinцелое числоn— количество строк архива. - Читает следующие
nстрок; каждая содержит два токена через пробел:order_id(целое число) иorder_date(строка видаYYYY-MM-DD). Пример строки:101 2024-03-15. - Создаёт
pd.DataFrameс именемdfсо столбцами'order_id'(int) и'order_date'(str). - Преобразует столбец
'order_date'вdatetime64[ns]черезpd.to_datetime. Сохраняет обратно вdf['order_date']. - Создаёт столбец
'year'черезdf['order_date'].dt.year, столбец'month'черезdf['order_date'].dt.month, столбец'weekday'черезdf['order_date'].dt.dayofweek. - Читает из
stdinцелое числоtarget_month— номер месяца для фильтрации (от 1 до 12). - Фильтрует строки, где
month == target_month, черезdf.loc[mask]. Сохраняет вfiltered. - Выводит в
stdoutровноlen(filtered) + 2строк:
- Строки 1…
len(filtered): для каждой строкиfiltered(в порядке исходного DataFrame) одну строку вида<order_id> <year> <month> <weekday>. - Предпоследняя строка:
count=<len(filtered)>. - Последняя строка:
weekdays=<список>, где список — значенияweekdayотфильтрованных строк через запятую и пробел, в порядке исходного DataFrame. Еслиfilteredпуст — выведитеweekdays=.
Ввод и вывод
Программа читает stdin и пишет в stdout.
Формат ввода
n order_id YYYY-MM-DD ... target_month
Дополнительные ограничения: n >= 1; таблица содержит явно описанные в условии столбцы.
Формат вывода
<order_id> <year> <month> <weekday>
...
count=<k>
weekdays=<w1>, <w2>, ...
Примеры
Пример 1
- Ввод
5 101 2024-01-08 102 2024-03-15 103 2024-03-22 104 2024-05-01 105 2024-03-01 3
- Ожидаемый результат
102 2024 3 4 103 2024 3 4 105 2024 3 4 count=3 weekdays=4, 4, 4
Попробуйте решить
РешениеPython
Без регистрации · результат не сохраняется
