将numpy数组列提取到字典中

我正在手动创建一个功能字典和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.read_csv()

我用这个代码来简化它:

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