2017-04-01 5 views
0

nehme an, ich habe einen Datenrahmen ähnlich wie diese:Wie ein Array aus zwei Spalten in Pandas erstellen

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 

Wie kann ich wählen col1 und col2 und sie in diesem Array?

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

Antwort

3

Sie können die zugrunde liegenden numpy Array über das .values Attribut zuzugreifen:

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

Sie können auch die gleiche Leistung mit dem folgenden Code erreichen.

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