Ключи словаря - это столбцы из информационного кадра, а значения - это тип объекта столбца.

Question

Напишите функцию, которая возвращает словарь, в котором ключами являются имена столбцов фрейма данных, а значениями являются «числовые», «категориальные», «порядковые», «дата / время» или «текстовые», соответствующиек типу объекта каждого столбца.

Мне удалось импортировать данные в виде фрейма данных. Затем я преобразовал df в словарь. Затем я получил ключи в качестве столбцов, но у меня возникли проблемы с назначением значений («числовые», «категориальные», «порядковые», «дата / время» или «текст») для каждого столбца.

# Importing the pandas package
import pandas as pd

def helper(df):
    # Creating data frame from csv.gz File. Enter the appropriate file path
    cobra = pd.read_csv(df, compression='gzip')
    cobra_dict = cobra.to_dict()
    dict_keys = cobra.keys()

    print(dict_keys)


helper(df)

Expected Results:
In [1]: helper(df[['offense_id','beat','x','y']])
Out[1]: {'beat': 'categorical',
         'offense_id': 'ordinal',
         'x': 'numeric',
         'y': 'numeric'}

Answer 1

IIUC,

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

def helper(df):
    # Creating data frame from csv.gz File. Enter the appropriate file path
    cobra = pd.read_csv(df, compression='gzip')
    cobra_dict = dict(zip(cobra.columns.tolist(),cobra .dtypes.tolist()))

    print(cobra_dict)

на моем собственном образце df.

print(my_dict)    
{'A': dtype('O'), 'B': dtype('int64'), 'C': dtype('float64')}