将熊猫数据转换为NumPy数组

Question

将熊猫数据转换为NumPy数组

我想知道如何将熊猫数据转换成NumPy数组。

数据：

import numpy as np

import pandas as pd

index = [1, 2, 3, 4, 5, 6, 7]

a = [np.nan, np.nan, np.nan, 0.1, 0.1, 0.1, 0.1]

b = [0.2, np.nan, 0.2, 0.2, 0.2, np.nan, np.nan]

c = [np.nan, 0.5, 0.5, np.nan, 0.5, 0.5, np.nan]

df = pd.DataFrame({'A': a, 'B': b, 'C': c}, index=index)

df = df.rename_axis('ID')

施予

label   A    B    C

ID                                 

1   NaN  0.2  NaN

2   NaN  NaN  0.5

3   NaN  0.2  0.5

4   0.1  0.2  NaN

5   0.1  0.2  0.5

6   0.1  NaN  0.5

7   0.1  NaN  NaN

我想将其转换为NumPy数组，如下所示：

array([[ nan,  0.2,  nan],

       [ nan,  nan,  0.5],

       [ nan,  0.2,  0.5],

       [ 0.1,  0.2,  nan],

       [ 0.1,  0.2,  0.5],

       [ 0.1,  nan,  0.5],

       [ 0.1,  nan,  nan]])

我该怎么做？

作为一种奖励，是否可以像这样保存dtype？

array([[ 1, nan,  0.2,  nan],

       [ 2, nan,  nan,  0.5],

       [ 3, nan,  0.2,  0.5],

       [ 4, 0.1,  0.2,  nan],

       [ 5, 0.1,  0.2,  0.5],

       [ 6, 0.1,  nan,  0.5],

       [ 7, 0.1,  nan,  nan]],

     dtype=[('ID', '<i4'), ('A', '<f8'), ('B', '<f8'), ('B', '<f8')])

或者类似的？

Answer 1

若要将熊猫dataframe(Df)转换为numpy ndarray，请使用以下代码：df.values array([[nan,&nbsp;0.2,&nbsp;nan], &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;[nan,&nbsp;nan,&nbsp;0.5], &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;[nan,&nbsp;0.2,&nbsp;0.5], &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;[0.1,&nbsp;0.2,&nbsp;nan], &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;[0.1,&nbsp;0.2,&nbsp;0.5], &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;[0.1,&nbsp;nan,&nbsp;0.5], &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;[0.1,&nbsp;nan,&nbsp;nan]])