将Dataframe转换为2d阵列

时间:2019-02-06 08:25:58

标签: python dataframe conv-neural-network convolution

我有一个尺寸(140000,22)尺寸的数据框。

我必须创建尺寸相等的二维数组,才能将其传递到卷积神经网络。

能否请您指导如何在此数据框上进行转换

2 个答案:

答案 0 :(得分:0)

您只需要在DataFrame上调用.values

例如,如果您的数据帧称为df,则可以将df.values传递到卷积神经网络。

答案 1 :(得分:0)

使用arr = df.to_numpy()

见: https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.to_numpy.html

它将创建一个二维数组,其中每一行都是内部数组。

[[row1],
 [row2],
 [row3]...
 [row n]]

要将其转换为每列是内部数组的二维数组,请使用转置:

arr = np.transpose(arr)

[[all_data_in_col_1],
 [all_data_in_col_2],...
 [all_data_in_col_n]]