Этот скрипт генерирует DataFrame, состоящий из одного столбца, и преобразует его в One-Hot представление без использования функции get_dummies
из библиотеки pandas
. Для более красивого и удобного вывода используется библиотека tabulate
.
- Импорт библиотек:
import pandas as pd import random from tabulate import tabulate
- Генерация исходного DataFrame
- Создается список
lst
, содержащий 10 строк "robot" и 10 строк "human". - Список перемешивается случайным образом.
- На основе списка создается
DataFrame data
с одним столбцомwhoAmI
.
- Преобразование в One-Hot представление
- Создается пустой
DataFrame data_one_hot
. - Для каждого уникального значения в столбце
whoAmI
создается новый столбец вdata_one_hot
. - В новый столбец записываются значения 1, если строка соответствует уникальному значению, и 0 в противном случае.
- Вывод DataFrame
- Для красивого форматирования вывода используется функция
tabulate
.
- Параметры tabulate
headers='keys'
: Использует имена столбцов DataFrame в качестве заголовков.tablefmt='pretty'
: Форматирует таблицу с границами и отступами для лучшей читаемости.