Извлекать столбцы массива в словарь

Question

Я создаю вручную словарь функций и массив numpy.

columns = ['a', 'b', 'c', 'd']
features = {'a': df_train.values[:, 0],
            'b': df_train.values[:, 1],
            'c': df_train.values[:, 2],
            'd': df_train.values[:, 3]}

df_train - это кадр данных Pandas, созданный в pandas.read_csv()

. Я использую этот код дляупростить это:

features = {c: df_train.values[:, i] for i, c in enumerate(columns)}

Есть ли более питонский путь?(например, с помощью почтового индекса?)

Answer 1

Вы можете использовать транспонирование:

dict(zip(columns, df_train.values.T))

или

{k: v for k, v in zip(columns, df_train.values.T)}