Pandas Базовый №1. Создание DataFrame и запись в CSV

Файлы к уроку:

• Для спонсоров Boosty
• Для спонсоров VK

• YouTube
• VK

Ссылки:

• Страница курса
• Плейлист YouTube
• Плейлист ВК

Описание

В этом уроке мы научимся вручную создавать объект DataFrame. Разберем 2 способа создания объекта DataFrame:

• Из словаря списка
• Из списка словарей

Также мы научимся записывать DataFrame в файл CSV.

Решение

Сначала нужно импортировать модули pandas и numpy.

# Импорт модулей
import pandas as pd
import numpy as np

Создадим DataFrame из словаря списков

Создадим списки, которые потом станут столбцами.

# Создадим списки, которые станут столбцами датафрейма
name = ['Rocka Rolla', 'Sad Wings of Destiny', 'Sin After Sin', 'Stained Class', 'Killing Machine']
year = [1974, 1976, 1977, 1978, 1978]

Создадим словарь.

# Создадим словарь
albums_dict = {'album': name,
             'release_year': year}

Теперь из этого словаря можно создать датафрейм.

# Создаем DataFrame
albums_df = pd.DataFrame(albums_dict)

Индексация и как ее изменить

По умолчанию создается Range индекс.

# По умолчанию индексация - это диапазон чисел
albums_df.index

Можно вручную изменить индекс.

# Можно задать пользовательскую индексацию
albums_df.index = ['a', 'b', 'c', 'd', 'e']

Создание DataFrame из списка словарей

По умолчанию столбцы идут в алфавитном порядке. С помощью параметра columns можно указать порядок столбцов.

albums_df_2 = pd.DataFrame([{'album': 'Point of Entry',
                             'year': 1981},
                            {'album': 'Screaming for Vengeance',
                             'year': 1982}],
                           columns=['year', 'album'])

Запишем результат в CSV

С помощью метода to_csv можно записать датафрейм в CSV файл.

Примененные функции

• pandas.DataFrame
• pandas.DataFrame.to_csv

Курс Pandas Базовый

Номер урока	Урок	Описание
1	Pandas Базовый №1. Создание DataFrame и запись в CSV	Познакомимся с объектом DataFrame. Научимся его создавать двумя разными способами и научимся записывать его в файл.
2	Pandas Базовый №2. Создание DataFrame 2	Изучим еще несколько способов создания объекта DataFrame. В этом уроке мы создадим DataFrame из массива numpy, Series, словаря Series.
3	Pandas Базовый №3. Отбор строк и столбцов, Размерность, Импорт CSV	Получить информацию о размере DataFrame, отбор строк и столбцов, индексация.
4	Pandas Базовый №4. Операции со столбцами DataFrame	Операции со столбцами в Pandas. Переименование столбцов, добавление новых столбцов, изменить существующий столбец, удаление столбцов.
5	Pandas Базовый №5. Операции со строками	Объединение по вертикали методами append и concat, Создание строк вручную, Удаление строк методом drop, Фильтрация строк условием или срезом.
6	Pandas Базовый №6. Индексы	Зачем нужны индексы, Как задать индекс, Как пользоваться индексами.
7	Pandas Базовый №7. Категории	Что такое категориальные переменные.