发布于 2年前

转换Pandas dataframe为numpy array数组

常用用于转换Pandas dataframe为numpy array有两种方法:

np_array = df.as_matrix(columns=None)
np_array = df.values

但这两种方法都有一个缺陷:丢失dtype和name。

另外一种可选的方法是使用df.to_records(),它会返回recarray,然后使用np.asarray()把转recarray转换为array。

ra = df.to_records(index=False)
np_array = np.asarray(ra)

它会保留dtype和name。

©2020 edoou.com   京ICP备16001874号-3