Как получить список имен столбцов в featuretools - PullRequest
Новая
       7

Как получить список имен столбцов в featuretools

0 голосов
/ 18 сентября 2018

Как я могу получить список имен столбцов в featuretools.

в фреймах данных pandas Я просто набираю этот код 

dataframe.columns

, который возвращает список имен столбцов

однако я попытался сделать это в наборе сущностей и потерпел неудачу.я должен преобразовать набор сущностей в фрейм данных?

Сейчас я делаю это путем преобразования переменных в строку и делаю регулярные выражения, чтобы получить имя переменной.Тем не менее, я считаю, что есть лучший способ сделать это.

спасибо,

1 Ответ

0 голосов
/ 18 сентября 2018

Набор объектов содержит несколько фреймов данных для каждого объекта.

In [1]: import featuretools as ft

In [2]: es = ft.demo.load_retail()

In [3]: es
Out[3]: 
Entityset: demo_retail_data
  Entities:
    order_products [Rows: 401604, Columns: 7]
    customers [Rows: 4372, Columns: 2]
    products [Rows: 3684, Columns: 3]
    orders [Rows: 22190, Columns: 5]
  Relationships:
    order_products.product_id -> products.product_id
    order_products.order_id -> orders.order_id
    orders.customer_name -> customers.customer_name

Если вы хотите, чтобы переменные для сущности "orders" запустились 

In [4]: es["orders"].variables
Out[4]: 
[<Variable: order_id (dtype = index)>,
 <Variable: country (dtype = categorical)>,
 <Variable: cancelled (dtype = boolean)>,
 <Variable: first_order_products_time (dtype: datetime_time_index, format: None)>,
 <Variable: customer_name (dtype = id)>]

Если вы хотите получить доступ к самому базовому фрейму данных, запустите 

In [5]: es["orders"].df.head(5)
Out[5]: 
       order_id         country  cancelled first_order_products_time  customer_name
536365   536365  United Kingdom      False       2010-12-01 08:26:00   Andrea Brown
536366   536366  United Kingdom      False       2010-12-01 08:28:00   Andrea Brown
536367   536367  United Kingdom      False       2010-12-01 08:34:00  Krista Maddox
536368   536368  United Kingdom      False       2010-12-01 08:34:00  Krista Maddox
536369   536369  United Kingdom      False       2010-12-01 08:35:00  Krista Maddox
...