supongamos que tengo un DataFrame similar a este:

d = {'col1': [0, 2, 4], 'col2': [1, 3, 5], 'col3': [2, 4, 8]}
df = pd.DataFrame(d)

   col1  col2  col3
0     0     1     2
1     2     3     4
2     4     5     8

¿Cómo puedo select col1 y col2 y convertirlos en esta matriz?

array([[0, 1],
       [2, 3],
       [4, 5]])

2 Comentarios

  1. 9

    Puede acceder a la subyacente numpy de la matriz a través de la .values atributo:

    df[['col1', 'col2']].values
    Out: 
    array([[0, 1],
           [2, 3],
           [4, 5]])
  2. 0

    Usted también puede lograr el mismo resultado con el siguiente código.

    import numpy as np
    np.array(df[['col1','col2']])
    Out[60]: 
    array([[0, 1],
           [2, 3],
           [4, 5]])

Dejar respuesta

Please enter your comment!
Please enter your name here