Supposons qu'une trame de données Pandas ressemble à:
X_test.head(4)
BoxRatio Thrust Velocity OnBalRun vwapGain
5 -0.163 -0.817 0.741 1.702 0.218
8 0.000 0.000 0.732 1.798 0.307
11 0.417 -0.298 2.036 4.107 1.793
13 0.054 -0.574 1.323 2.553 1.185
Comment puis-je extraire la troisième ligne (comme row3) en tant que trame de données pd? En d'autres termes, row3.shape doit être (1,5) et row3.head () doit être:
0.417 -0.298 2.036 4.107 1.793
Utilisez .iloc
Avec des crochets doubles pour extraire un DataFrame, ou des crochets simples pour extraire une série.
>>> import pandas as pd
>>> df = pd.DataFrame({'col1': [1, 2], 'col2': [3, 4]})
>>> df
col1 col2
0 1 3
1 2 4
>>> df.iloc[[1]] # DataFrame result
col1 col2
1 2 4
>>> df.iloc[1] # Series result
col1 2
col2 4
Name: 1, dtype: int64
Ceci s'étend à d'autres formes d'indexation DataFrame également, à savoir .loc
Et .__getitem__()
:
>>> df.loc[:, ['col2']]
col2
0 3
1 4
>>> df[['col2']]
col2
0 3
1 4